Searched refs:concat_dimension (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_constant_folding_test.cc106 int concat_dimension; member in struct:xla::__anon25617::TestConfig
121 dimensions[test_config.concat_dimension] = csize;
128 dimensions[test_config.concat_dimension] = concat_size;
131 shape, operands, test_config.concat_dimension));
/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_op_test.cc39 // in size, and concat them together along "concat_dimension"
41 static void ConcatHelper(int iters, int concat_dimension, int dim2) { argument
48 concat_dim.scalar<int32>()() = concat_dimension;
92 static void ConcatManyHelper(int iters, int concat_dimension, int dim2) { argument
100 concat_dim.scalar<int32>()() = concat_dimension;
129 static void MemcpyAlternativeHelper(int iters, int concat_dimension, int dim2) { argument
H A Dquantized_concat_op_test.cc247 // in size, and concat them together along "concat_dimension".
251 static void ConcatHelper(int iters, int concat_dimension, bool same_limits, argument
260 Tensor concat_dim = test::AsScalar<int32>(concat_dimension);
289 ConcatHelper<qint32>(iters, 0 /* concat_dimension */, true /* same_limits */,
294 ConcatHelper<qint32>(iters, 1 /* concat_dimension */, true /* same_limits */,
299 ConcatHelper<qint32>(iters, 0 /* concat_dimension */, false /* same_limits */,
304 ConcatHelper<qint32>(iters, 1 /* concat_dimension */, false /* same_limits */,
314 ConcatHelper<qint32>(iters, 0 /* concat_dimension */, true /* same_limits */,
319 ConcatHelper<qint32>(iters, 1 /* concat_dimension */, true /* same_limits */,
324 ConcatHelper<qint32>(iters, 0 /* concat_dimension */, fals
[all...]
/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
H A Dops.py201 concat_dimension = axis_names.index(axis_name)
202 concat_tensor = array_ops.concat(tensors, concat_dimension, name=scope)
204 concat_axes = (values[:concat_dimension] + [concat_axis] +
205 values[concat_dimension + 1:])
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconcat_test.cc483 int64 concat_dimension; member in struct:xla::__anon25729::R2BinarySpec
501 builder.ConcatInDim({a0, a1}, spec.concat_dimension);
504 ReferenceUtil::Concat2D(lhs, rhs, spec.concat_dimension);

Completed in 566 milliseconds