Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_reshard_variables_op_util.cc226 absl::Span<const TensorShapeProto* const> output_tensor_shape_protos, in UpdateOutputVariables() argument
234 if (sub_elements != output_tensor_shape_protos.size()) { in UpdateOutputVariables()
237 output_tensor_shape_protos.size()); in UpdateOutputVariables()
250 TensorShape::IsValidShape(*output_tensor_shape_protos[i])); in UpdateOutputVariables()
251 TensorShape shape(*output_tensor_shape_protos[i]); in UpdateOutputVariables()
H A Dtpu_execute_op.cc395 absl::Span<const TensorShapeProto* const> output_tensor_shape_protos, in AllocateOutputTensors() argument
405 if (sub_elements != output_tensor_shape_protos.size()) { in AllocateOutputTensors()
408 output_tensor_shape_protos.size()); in AllocateOutputTensors()
418 TensorShape::IsValidShape(*output_tensor_shape_protos[i])); in AllocateOutputTensors()
419 TensorShape shape(*output_tensor_shape_protos[i]); in AllocateOutputTensors()
H A Dtpu_reshard_variables_op_util.h52 absl::Span<const TensorShapeProto* const> output_tensor_shape_protos,