Home
last modified time | relevance | path

Searched defs:new_tensor (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAutogradComposite.cpp52 auto new_tensor = at::zeros_symint({other_storage_numel}, other.options()); in _new_zeros_with_same_feature_meta() local
86 auto new_tensor = at::zeros_symint({storage_numel}, other.options()); in _new_zeros_with_same_feature_meta() local
H A DComplexHelper.h26 auto new_tensor = detail::make_tensor<TensorImpl>( in view_tensor() local
H A DResize.cpp307 auto new_tensor = at::empty(src_tensor.sizes(), src_tensor.options()); in resize_bytes_nocuda() local
H A DTensorConversions.cpp788 auto new_tensor = detail::make_tensor<TensorImpl>( in view_dtype() local
/aosp_15_r20/external/executorch/extension/tensor/test/
H A Dtensor_ptr_test.cpp350 auto new_tensor = make_tensor_ptr(*tensor); in TEST_F() local
395 auto new_tensor = make_tensor_ptr(*tensor); in TEST_F() local
440 auto new_tensor = make_tensor_ptr(*tensor); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.cc484 std::unique_ptr<Tensor> new_tensor = forward_input( in forward_input_to_output_with_shape() local
584 std::unique_ptr<Tensor> new_tensor = in forward_input_or_allocate_temp() local
754 Tensor new_tensor( in allocate_tensor() local
947 auto new_tensor = MakeUnique<Tensor>(); in maybe_set_output_by_allocate_and_copy() local
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dutils.h47 at::Tensor* new_tensor = new at::Tensor(std::move(tensor)); in new_tensor_handle() local
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dtensor_new.cpp1470 auto new_tensor = internal_new_from_data( in tensor_ctor() local
1510 Tensor new_tensor( in new_tensor() function
1534 auto new_tensor = new_from_data_copy( in new_tensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcustom_device_op_handler.cc81 tensorflow::ImmediateExecutionTensorHandle* new_tensor; in Execute() local
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_assign.cpp238 Tensor<int, 1> new_tensor(5); in test_auto_resize() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dindexed_array_analysis.h332 T* new_tensor = new T(std::forward<Args>(args)...); in Construct() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dutils.cc34 TfLiteTensor** new_tensor, in CreateNewTensorWithDifferentType()
H A Dutils_test.cc59 TfLiteTensor* new_tensor = nullptr; in TEST() local
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/
H A DClWorkloadRuntime.cpp103 auto new_tensor = inserted_pair->second.get(); in add_aux_tensor() local
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Doptim.cpp79 torch::Tensor new_tensor) { in assign_parameter()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc1514 TfLiteTensor* new_tensor = &context_->tensors[*tensor_index]; in AddNewInputConstantTensor() local
1887 TfLiteTensor* new_tensor = &context_->tensors[new_tensor_index]; in AddTensor() local
1917 TfLiteTensor* new_tensor = &context_->tensors[new_tensor_index]; in AddTensor() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_serialization.py451 def __init__(self, new_tensor): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMath.cpp56 Tensor new_tensor = at::constant_pad_nd(t, IntArrayRef(padd), value); in pad_tensor_to_shape() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp92 void OperandInfo::exchange_tensor(c10::MaybeOwned<TensorBase> &&new_tensor) { in exchange_tensor()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model.cc105 std::unique_ptr<TensorT> new_tensor(new TensorT); in DuplicateBiasesWithMultipleUses() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer.cc259 const TensorId new_tensor = ParseTensorName(new_input); in UpdateConsumers() local