Searched refs:output_tensor_shape_proto (Results 1 – 1 of 1) sorted by relevance
156 TensorShapeProto output_tensor_shape_proto = in CompileToCustomCallCallingTfKernel() local158 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()