Home
last modified time | relevance | path

Searched defs:partition_dim (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ops/
H A Dtpu_partitioned_input_op.cc27 const int partition_dim) { in _UpdatePartitionDim()
45 int partition_dim; in __anon5fb8c9c80102() local
H A Dtpu_partitioned_output_op.cc35 int partition_dim; in __anon0352b5690102() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_reorder_replicate_and_partitioned_inputs.cc46 int64_t partition_dim = first_partitioned_input.partition_dim(); in ReorderReplicateAndPartitionedInputs() local