Searched defs:CreateKernelTensor (Results 1 – 7 of 7) sorted by relevance
964 void DeviceAddressUtils::CreateKernelTensor(const device::DeviceAddressPtr &device_address, in CreateKernelTensor() function in mindspore::runtime::DeviceAddressUtils980 void DeviceAddressUtils::CreateKernelTensor(const ValuePtr &input_value) { in CreateKernelTensor() function in mindspore::runtime::DeviceAddressUtils992 void DeviceAddressUtils::CreateKernelTensor(const tensor::TensorPtr &input_tensor) { in CreateKernelTensor() function in mindspore::runtime::DeviceAddressUtils1001 void DeviceAddressUtils::CreateKernelTensor(const device::DeviceAddressPtr &device_address, in CreateKernelTensor() function in mindspore::runtime::DeviceAddressUtils
69 KernelTensor *CreateKernelTensor(const std::vector<int64_t> &shape, const TypePtr &dtype) { in CreateKernelTensor() function in mindspore::kernel::SparseApplyProximalAdagradCpuKernelTest
70 KernelTensor *CreateKernelTensor(const std::vector<int64_t> &shape, const TypePtr &dtype) { in CreateKernelTensor() function in mindspore::kernel::FusedSparseFtrlCpuKernelTest
74 KernelTensor *CreateKernelTensor(const std::vector<int64_t> &shape, const TypePtr &dtype) { in CreateKernelTensor() function in mindspore::kernel::SparseApplyAdamCpuKernelTest
72 KernelTensor *CreateKernelTensor(const std::vector<int64_t> &shape, const TypePtr &dtype) { in CreateKernelTensor() function in mindspore::kernel::SparseApplyLazyAdamCpuKernelTest
601 void CreateKernelTensor(const std::vector<std::vector<tensor::TensorPtr>> &input_tensors, in CreateKernelTensor() function623 void CreateKernelTensor(const BaseRef &arg) { in CreateKernelTensor() function642 void CreateKernelTensor(const VectorRef &args) { in CreateKernelTensor() function
83 KernelTensorPtr CreateKernelTensor(const abstract::AbstractBasePtr &cur_abstract, const TypeId &rea… in CreateKernelTensor() function