Searched refs:scalarToTensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | TensorIndexing.cpp | 54 value = at::indexing::scalarToTensor(v, device(kCPU).dtype(kFloat), at::Device(kCPU)); in set_item()
56 value = at::indexing::scalarToTensor(v, self.options(), at::Device(kCPU)); in set_item()
58 value = at::indexing::scalarToTensor(v, self.options(), self_device); in set_item()
|
| H A D | TensorIndexing.h | 377 inline Tensor scalarToTensor( in scalarToTensor() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| H A D | type_conversion.cc | 71 llvm::Optional<Value> scalarToTensor(OpBuilder& builder, Type /*type*/, in scalarToTensor() function
98 addArgumentMaterialization(scalarToTensor); in LinalgTypeConverter()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | python_variable_indexing.cpp | 137 at::indexing::scalarToTensor(scalar, options, device)); in valueToTensor()
139 return at::indexing::scalarToTensor(scalar, options, device); in valueToTensor()
|