Searched defs:concat_dimension (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_cross_replica_ops.cc | 41 int concat_dimension; in __anonc4e6eef40102() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_op_test.cc | 61 int concat_dimension, int dim2, in ConcatHelper()
159 int concat_dimension, int dim2) { in ConcatManyHelper()
|
| D | quantized_concat_op_test.cc | 252 int concat_dimension, bool same_limits, int dim2) { in ConcatHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | xla_sharding_util.cc | 102 mlir::LogicalResult CreateConcatOp(const int concat_dimension, in CreateConcatOp()
426 int concat_dimension = sharding.tile_assignment_dimensions_size() - 1; in HandleTileShardedOutputs() local
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 46 concat_dimension, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_constant_folding_test.cc | 113 int concat_dimension; in TEST_F() member
|
| D | shape_inference.cc | 2062 const Shape& shape, int64 split_dimension, int64 concat_dimension, in InferAllToAllShape()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 773 int64 concat_dimension; member
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2858 int64 concat_dimension, int64 split_count, in AllToAll()
2872 XlaOp operand, int64 split_dimension, int64 concat_dimension, in AllToAllArray()
2924 int64 concat_dimension, int64 split_count, in AllToAllTuple()
4678 int64 concat_dimension, int64 split_count, in AllToAll()
4686 int64 concat_dimension, int64 split_count, in AllToAllTuple()
|