Home
last modified time | relevance | path

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

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