Searched defs:CastTensor (Results 1 – 5 of 5) sorted by relevance
571 std::optional<tensor::BaseTensorPtr> PyBoostUtils::CastTensor(const std::optional<tensor::BaseTenso… in CastTensor() function in mindspore::kernel::pyboost::PyBoostUtils585 tensor::BaseTensorPtr PyBoostUtils::CastTensor(const tensor::BaseTensorPtr &tensor, const TypeId &t… in CastTensor() function in mindspore::kernel::pyboost::PyBoostUtils595 std::vector<tensor::BaseTensorPtr> PyBoostUtils::CastTensor(const std::vector<tensor::BaseTensorPtr… in CastTensor() function in mindspore::kernel::pyboost::PyBoostUtils609 std::vector<tensor::BaseTensorPtr> PyBoostUtils::CastTensor(const std::vector<tensor::BaseTensorPtr… in CastTensor() function in mindspore::kernel::pyboost::PyBoostUtils
141 Tensor *CastTensor(Tensor *tensor, TypeId dst_data_type, bool support_fp16) { in CastTensor() function
144 AnfNodePtr ConvertBFloat16::CastTensor(const ValueNodePtr &value_node) { in CastTensor() function in mindspore::graphkernel::ConvertBFloat16
57 nvinfer1::ITensor *DecoderTensorRT::CastTensor(TensorRTContext *ctx, const TensorInfo &ms_tensor, in CastTensor() function in mindspore::lite::DecoderTensorRT
85 nvinfer1::ITensor *EncoderTensorRT::CastTensor(TensorRTContext *ctx, const TensorInfo &ms_tensor, in CastTensor() function in mindspore::lite::EncoderTensorRT