Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc556 bool xla_spmd_input_sharded = false; in FindTpuReplicatedInputAndXlaSharding() local
749 int32_t minimum_input_tensors_packing, bool xla_spmd_input_sharded, in CreateConcatAndSplitNodesForInputTensor()
2207 bool xla_spmd_input_sharded = false; in OptimizeTpuInputOutputTensors() local