| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_reduction_sycl.cpp | 69 using scalar_tensor = Tensor<DataType, 0, DataLayout, IndexType>; in test_full_reductions_sum_with_offset_sycl() typedef
143 using scalar_tensor = Tensor<DataType, 0, DataLayout, IndexType>; in test_full_reductions_max_with_offset_sycl() typedef
276 using scalar_tensor = Tensor<DataType, 0, DataLayout, IndexType>; in test_full_reductions_mean_with_offset_sycl() typedef
321 using scalar_tensor = Tensor<DataType, 0, DataLayout, IndexType>; in test_full_reductions_mean_with_odd_offset_sycl() typedef
399 using scalar_tensor = Tensor<DataType, 0, DataLayout, IndexType>; in test_full_reductions_min_with_offset_sycl() typedef
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Bucketization.cpp | 199 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_out_cpu() local
224 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_cpu() local
|
| H A D | TensorFactories.cpp | 842 Tensor scalar_tensor(const Scalar& s, in scalar_tensor() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Bucketization.cu | 186 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_out_cuda() local
211 const Tensor& scalar_tensor = searchsorted_scalar_tensor(self, sorted_sequence.device()); in searchsorted_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesBinaryOps.cpp | 337 #define BINARY_SCALAR_3(op, tensor_tensor, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
342 #define BINARY_SCALAR_3_Tensor(op, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | fill_ops.cc | 220 ITensorProxyPtr scalar_tensor = nullptr; in Convert() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | generate_box_proposals_op.cu.cc | 299 const Tensor& scalar_tensor = context->input(input); in GetScalarValue() local
|
| /aosp_15_r20/external/executorch/runtime/core/exec_aten/util/ |
| H A D | tensor_util.h | 402 #define ET_EXTRACT_SCALAR_TENSOR(scalar_tensor, out_val) \ argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tensor_array_ops_decomposition.cc | 217 tensorflow::Tensor scalar_tensor(tensorflow::DT_FLOAT, {}); in HandleTensorArrayV3Op() local
|
| /aosp_15_r20/external/pytorch/torch/_prims/ |
| H A D | __init__.py | 2763 scalar_tensor = _make_prim( variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/ir/ |
| H A D | ir.cpp | 160 auto scalar_tensor = tensor.view(std::vector<int64_t>{}).item(); in printAttribute() local
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | __init__.py | 5550 def scalar_tensor( function
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 3446 def scalar_tensor(g: jit_utils.GraphContext, scalar, dtype, *options): function
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 6046 def scalar_tensor(s, dtype=None, layout=None, device=None, pin_memory=None): function
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 87 torch::Tensor scalar_tensor = torch::scalar_tensor( in TEST_F() local
11436 torch::Tensor scalar_tensor = torch::scalar_tensor( in TEST_F() local
|