| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 41 int concat_dimension; in __anonb72bce3d0102() local
44 TF_RETURN_IF_ERROR(c->GetAttr("concat_dimension", &concat_dimension)); in __anonb72bce3d0102()
46 if (concat_dimension < 0 || concat_dimension >= rank) { in __anonb72bce3d0102()
47 return errors::InvalidArgument("concat_dimension ", concat_dimension, in __anonb72bce3d0102()
62 if (i == concat_dimension) { in __anonb72bce3d0102()
65 in_idx = concat_dimension; in __anonb72bce3d0102()
|
| D | ops.pbtxt | 695 name: "concat_dimension"
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 45 concat_dimension, argument
71 concat_dimension=concat_dimension,
86 concat_dimension=op.get_attr("split_dimension"),
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_AllToAll.pbtxt | 31 name: "concat_dimension"
54 blocks along `concat_dimension` as the output.
61 concat_dimension=0
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_op_test.cc | 41 static void ConcatHelper(int iters, int concat_dimension, int dim2) { in ConcatHelper() argument
48 concat_dim.scalar<int32>()() = concat_dimension; in ConcatHelper()
96 static void ConcatManyHelper(int iters, int concat_dimension, int dim2) { in ConcatManyHelper() argument
104 concat_dim.scalar<int32>()() = concat_dimension; in ConcatManyHelper()
133 static void MemcpyAlternativeHelper(int iters, int concat_dimension, int dim2) { in MemcpyAlternativeHelper() argument
|
| D | quantized_concat_op_test.cc | 251 static void ConcatHelper(int iters, int concat_dimension, bool same_limits, in ConcatHelper() argument
260 Tensor concat_dim = test::AsScalar<int32>(concat_dimension); in ConcatHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_constant_folding_test.cc | 111 int concat_dimension; in TEST_F() member
126 dimensions[test_config.concat_dimension] = csize; in TEST_F()
133 dimensions[test_config.concat_dimension] = concat_size; in TEST_F()
136 shape, operands, test_config.concat_dimension)); in TEST_F()
|
| D | shape_inference.h | 135 int64 concat_dimension,
|
| D | shape_inference.cc | 1992 const Shape& shape, int64 split_dimension, int64 concat_dimension, in InferAllToAllShape() argument
2000 if (concat_dimension >= shape.rank() || concat_dimension < 0) { in InferAllToAllShape()
2003 concat_dimension, ShapeUtil::HumanString(shape)); in InferAllToAllShape()
2014 new_dimensions[concat_dimension] *= split_count; in InferAllToAllShape()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 527 int64 concat_dimension; member
545 ConcatInDim(&builder, {a0, a1}, spec.concat_dimension); in TEST_P()
548 ReferenceUtil::Concat2D(lhs, rhs, spec.concat_dimension); in TEST_P()
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 202 concat_dimension = axis_names.index(axis_name)
203 concat_tensor = array_ops.concat(tensors, concat_dimension, name=scope)
205 concat_axes = (values[:concat_dimension] + [concat_axis] +
206 values[concat_dimension + 1:])
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 499 const LocalOp& operand, int64 split_dimension, int64 concat_dimension, in AllToAll() argument
506 return xla::AllToAll(operand.op(), split_dimension, concat_dimension, in AllToAll()
|
| D | local_computation_builder.h | 256 int64 concat_dimension, int64 split_count,
|
| D | xla_client.py | 1208 concat_dimension, argument
1238 return self._client.AllToAll(operand, split_dimension, concat_dimension,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 475 int64 concat_dimension, int64 split_count,
891 int64 concat_dimension, int64 split_count,
1594 int64 concat_dimension, int64 split_count,
|
| D | xla_builder.cc | 2160 int64 concat_dimension, int64 split_count, in AllToAll() argument
2174 concat_dimension, split_count) in AllToAll()
2210 return this->ConcatInDim(received, concat_dimension); in AllToAll()
3245 int64 concat_dimension, int64 split_count, in AllToAll() argument
3247 return operand.builder()->AllToAll(operand, split_dimension, concat_dimension, in AllToAll()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 44 `concat_dimension`.
62 <b> `AllToAll(operand, split_dimension, concat_dimension, split_count,
73 | `concat_dimension` | `int64` | a value in the interval `[0, |
92 AllToAll(x, /*split_dimension=*/1, /*concat_dimension=*/0, /*split_count=*/4);
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 85 …argspec: "args=[\'input\', \'group_assignment\', \'concat_dimension\', \'split_dimension\', \'spli…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 85 …argspec: "args=[\'input\', \'group_assignment\', \'concat_dimension\', \'split_dimension\', \'spli…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 25654 func AllToAll(scope *Scope, input tf.Output, group_assignment tf.Output, concat_dimension int64, sp…
25658 …attrs := map[string]interface{}{"concat_dimension": concat_dimension, "split_dimension": split_dim…
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 1574 name: "concat_dimension"
1627 name: "concat_dimension"
|
| D | ops_history.v1.pbtxt | 1574 name: "concat_dimension"
1627 name: "concat_dimension"
|