| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesLinearAlgebra.cpp | 188 const auto tensor_ = check_and_reshape_input(tensor, batch_dim); in apply_one() local
196 const auto tensor_ = check_and_reshape_input(tensor, batch_dim); in apply_two() local
205 const auto tensor_ = check_and_reshape_input(tensor, batch_dim); in apply_three() local
214 const auto tensor_ = check_and_reshape_input(tensor, batch_dim); in apply_four() local
240 const auto tensor_ = std::get<0>(tensor_other); in apply_one() local
250 const auto tensor_ = std::get<0>(tensor_other); in apply_two() local
342 const auto tensor_ = std::get<0>(tensor_other); in cholesky_solve_batch_rule() local
|
| H A D | BatchRulesBinaryOps.cpp | 24 auto tensor_ = std::get<0>(tensor_other); in _binary_pointwise_batch_rule() local
109 auto tensor_ = moveBatchDimToFront(tensor, tensor_batch_dim); in binary_pointwise_inplace_batch_rule() local
131 auto tensor_ = moveBatchDimToFront(tensor, tensor_batch_dim); in comparison_pointwise_batch_rule() local
|
| H A D | LegacyVmapTransforms.h | 157 Tensor tensor_; member
|
| H A D | BatchRulesHelper.cpp | 180 auto tensor_ = moveBatchDimToFront(tensor, tensor_batch_dim); in _binary_pointwise_helper() local
|
| /aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/ |
| H A D | converters.cc | 88 std::unique_ptr<tf::Tensor> tensor_; member in fcp::aggregation::tensorflow::NumericTensorDataAdapter
110 std::unique_ptr<tf::Tensor> tensor_; member in fcp::aggregation::tensorflow::StringTensorDataAdapter
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | ProcessGroupGloo.hpp | 200 at::Tensor tensor_; member in c10d::ProcessGroupGloo::SendWork
223 at::Tensor tensor_; member in c10d::ProcessGroupGloo::RecvWork
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device_util.h | 333 : tensor_(std::move(tensor)), in tensor_() function
340 std::unique_ptr<parallel_device::ParallelTensor> tensor_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ |
| H A D | tensor_impl.h | 56 LazyTensorPtr tensor_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | queue_ops.cc | 93 Tensor tensor_; member in tensorflow::FakeQueueOp
|
| H A D | variable_ops.cc | 63 Tensor tensor_; member in tensorflow::LegacyVar
|
| H A D | sparse_cross_op.cc | 193 const Tensor& tensor_; member in tensorflow::__anonad3ebfaf0111::DenseTensorColumn
216 const Tensor& tensor_; member in tensorflow::__anonad3ebfaf0111::KeyedDenseTensorColumn
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | tensor_coding.h | 105 Tensor tensor_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | fallback_tensor.h | 44 tensorflow::Tensor tensor_; variable
|
| H A D | fallback_tensor.cc | 50 tensorflow::Tensor tensor_; member in tensorflow::tfrt_stub::__anonb0533b900111::ImmutableTensorBuffer
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | resource_var.h | 108 Tensor tensor_; variable
|
| /aosp_15_r20/external/executorch/backends/qualcomm/aot/wrappers/ |
| H A D | TensorWrapper.h | 113 Qnn_Tensor_t tensor_ = QNN_TENSOR_INIT; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | LegacyVmapTransforms.h | 153 Tensor tensor_; member
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | extension.cpp | 24 torch::Tensor tensor_; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/internal/ |
| H A D | tfprof_tensor.h | 166 std::unique_ptr<Tensor> tensor_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/experimental/base/public/ |
| H A D | tensor.h | 124 std::unique_ptr<TF_Tensor, TFTensorDeleter> tensor_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | weights.h | 135 Tensor tensor_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| H A D | binary_ops.cc | 152 std::array<ITensorProxyPtr, 2> tensor_{nullptr, nullptr}; member in tensorflow::tensorrt::convert::__anon4e53edec0111::ConvertBinaryImpl
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | list_model_parameters.cpp | 120 auto tensor_ = attr.toTensor(); in getParamAttributes() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/detail/ |
| H A D | TensorDataContainer.h | 351 at::Tensor tensor_; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | Formatting.cpp | 258 std::ostream& print(std::ostream& stream, const Tensor & tensor_, int64_t linesize) { in print()
|