Home
last modified time | relevance | path

Searched refs:minimum_input_tensors_packing (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc749 int32_t minimum_input_tensors_packing, bool xla_spmd_input_sharded, in CreateConcatAndSplitNodesForInputTensor() argument
775 if (num_tensors < minimum_input_tensors_packing) { in CreateConcatAndSplitNodesForInputTensor()
2219 runtime_params_.minimum_input_tensors_packing > 1 || in OptimizeTpuInputOutputTensors()
2265 if (runtime_params_.minimum_input_tensors_packing > 1) { in OptimizeTpuInputOutputTensors()
2273 runtime_params_.minimum_input_tensors_packing, in OptimizeTpuInputOutputTensors()
H A Dtpu_functional_ops.h67 int32_t minimum_input_tensors_packing, bool xla_spmd_input_sharded,
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h110 int32_t minimum_input_tensors_packing; member