Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_executable_info.proto53 repeated TensorShapeProto output_tensor_shapes = 8; field
H A Dtpu_reshard_variables_op_util.cc246 std::vector<TensorShape> output_tensor_shapes; in UpdateOutputVariables() local
H A Dtpu_execute_op.cc414 std::vector<TensorShape> output_tensor_shapes; in AllocateOutputTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc505 std::vector<TensorShape> output_tensor_shapes; in PopulateOutputs() local