Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor.cc179 void ConvertToTensorShapeProto(ArrayRef<int64_t> shape, in ConvertToTensorShapeProto() function
194 ConvertToTensorShapeProto(tensor_type.getShape(), &tensor_shape_proto); in ConvertTypeToTensorShape()
315 ConvertToTensorShapeProto(shape, output->mutable_tensor_shape()); in ConvertToTensorProto()
Dconvert_tensor.h42 void ConvertToTensorShapeProto(llvm::ArrayRef<int64_t> shape,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_rewrite_pass.cc256 ConvertToTensorShapeProto(ranked_tensor_type.getShape(), &shape_proto); in SetMetadataProtoArgs()