Searched defs:concat_dimension (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | concat_op_test.cc | 61 int concat_dimension, int dim2, in ConcatHelper()
159 int concat_dimension, int dim2) { in ConcatManyHelper()
|
| H A D | quantized_concat_op_test.cc | 252 int concat_dimension, bool same_limits, int dim2) { in ConcatHelper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | xla_sharding_util.cc | 98 mlir::TF::ConcatOp CreateConcatOp(const int concat_dimension, in CreateConcatOp()
440 int concat_dimension = sharding.tile_assignment_dimensions_size() - 1; in HandleTileShardedOutputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | tpu_cross_replica_ops.cc | 43 int concat_dimension; in __anon173a7a2a0102() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ops/ |
| H A D | tpu_ops.py | 41 concat_dimension, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_constant_folding_test.cc | 114 int concat_dimension; in TEST_F() member
|
| H A D | shape_inference.cc | 2138 const Shape& shape, int64_t split_dimension, int64_t concat_dimension, in InferAllToAllShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 3078 int64_t concat_dimension, int64_t split_count, in AllToAll()
3092 int64_t concat_dimension, int64_t split_count, in AllToAllArray()
3184 int64_t concat_dimension, int64_t split_count, in AllToAllTuple()
4691 int64_t concat_dimension, int64_t split_count, in AllToAll()
4706 int64_t concat_dimension, int64_t split_count, in AllToAllTuple()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | concat_test.cc | 773 int64_t concat_dimension; member
|