Searched refs:OptimizeTpuInputOutputTensors (Results 1 – 2 of 2) sorted by relevance
206 Status OptimizeTpuInputOutputTensors(
1289 OptimizeTpuInputOutputTensors(graph.get(), enable_spmd_xla_partitioning, in ComputeAsync()2181 Status TPUPartitionedCallOp::OptimizeTpuInputOutputTensors( in OptimizeTpuInputOutputTensors() function in tensorflow::TPUPartitionedCallOp