Searched defs:concat_dimension (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 41 int concat_dimension; in __anonb72bce3d0102() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_op_test.cc | 41 static void ConcatHelper(int iters, int concat_dimension, int dim2) { in ConcatHelper()
96 static void ConcatManyHelper(int iters, int concat_dimension, int dim2) { in ConcatManyHelper()
133 static void MemcpyAlternativeHelper(int iters, int concat_dimension, int dim2) { in MemcpyAlternativeHelper()
|
| D | quantized_concat_op_test.cc | 251 static void ConcatHelper(int iters, int concat_dimension, bool same_limits, in ConcatHelper()
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 45 concat_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_constant_folding_test.cc | 111 int concat_dimension; in TEST_F() member
|
| D | shape_inference.cc | 1992 const Shape& shape, int64 split_dimension, int64 concat_dimension, in InferAllToAllShape()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 499 const LocalOp& operand, int64 split_dimension, int64 concat_dimension, in AllToAll()
|
| D | xla_client.py | 1208 concat_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 527 int64 concat_dimension; member
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2160 int64 concat_dimension, int64 split_count, in AllToAll()
3245 int64 concat_dimension, int64 split_count, in AllToAll()
|