Searched refs:concat_dimension (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_constant_folding_test.cc | 106 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 D | concat_op_test.cc | 39 // 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 D | quantized_concat_op_test.cc | 247 // 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 D | ops.py | 201 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 D | concat_test.cc | 483 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