Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/
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
DComplexHelper.h26 auto new_tensor = detail::make_tensor<TensorImpl>( in view_tensor() local
DResize.cpp307 auto new_tensor = at::empty(src_tensor.sizes(), src_tensor.options()); in resize_bytes_nocuda() local
DTensorConversions.cpp788 auto new_tensor = detail::make_tensor<TensorImpl>( in view_dtype() local
/external/pytorch/torch/csrc/inductor/aoti_torch/
Dutils.h47 at::Tensor* new_tensor = new at::Tensor(std::move(tensor)); in new_tensor_handle() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device_op_handler.cc81 tensorflow::ImmediateExecutionTensorHandle* new_tensor; in Execute() local
/external/eigen/unsupported/test/
Dcxx11_tensor_assign.cpp238 Tensor<int, 1> new_tensor(5); in test_auto_resize() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.h332 T* new_tensor = new T(std::forward<Args>(args)...); in Construct() local
/external/tensorflow/tensorflow/lite/delegates/
Dutils.cc34 TfLiteTensor** new_tensor, in CreateNewTensorWithDifferentType()
Dutils_test.cc59 TfLiteTensor* new_tensor = nullptr; in TEST() local
/external/executorch/extension/tensor/test/
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
/external/ComputeLibrary/src/dynamic_fusion/runtime/gpu/cl/
DClWorkloadRuntime.cpp103 auto new_tensor = inserted_pair->second.get(); in add_aux_tensor() local
/external/pytorch/torch/csrc/utils/
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
/external/tensorflow/tensorflow/core/framework/
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
/external/pytorch/test/cpp/api/
Doptim.cpp79 torch::Tensor new_tensor) { in assign_parameter()
/external/pytorch/test/
Dtest_serialization.py451 def __init__(self, new_tensor): argument
/external/pytorch/aten/src/ATen/native/nested/
DNestedTensorMath.cpp56 Tensor new_tensor = at::constant_pad_nd(t, IntArrayRef(padd), value); in pad_tensor_to_shape() local
/external/pytorch/aten/src/ATen/
DTensorIterator.cpp92 void OperandInfo::exchange_tensor(c10::MaybeOwned<TensorBase> &&new_tensor) { in exchange_tensor()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
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
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc105 std::unique_ptr<TensorT> new_tensor(new TensorT); in DuplicateBiasesWithMultipleUses() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc259 const TensorId new_tensor = ParseTensorName(new_input); in UpdateConsumers() local