Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dlight_outside_compilation.cc156 TensorShapeProto output_tensor_shape_proto = in CompileToCustomCallCallingTfKernel() local
158 if (output_tensor_shape_proto.unknown_rank()) { in CompileToCustomCallCallingTfKernel()
163 int rank = output_tensor_shape_proto.dim_size(); in CompileToCustomCallCallingTfKernel()
168 for (int d = 0; d < output_tensor_shape_proto.dim_size(); ++d) { in CompileToCustomCallCallingTfKernel()
169 auto* dim = output_tensor_shape_proto.mutable_dim(d); in CompileToCustomCallCallingTfKernel()
191 output_tensor_shape_proto; in CompileToCustomCallCallingTfKernel()
196 TensorShape::BuildTensorShape(output_tensor_shape_proto)); in CompileToCustomCallCallingTfKernel()