Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.h73 int32_t minimum_output_tensors_packing);
H A Dtpu_functional_ops.cc927 int32_t minimum_output_tensors_packing) { in CreateConcatAndSplitNodesForOutputTensor() argument
950 if (num_tensors < minimum_output_tensors_packing) { in CreateConcatAndSplitNodesForOutputTensor()
2248 if (runtime_params_.minimum_output_tensors_packing > 1) { in OptimizeTpuInputOutputTensors()
2262 shape_to_output, runtime_params_.minimum_output_tensors_packing)); in OptimizeTpuInputOutputTensors()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_ops_c_api.h111 int32_t minimum_output_tensors_packing; member