Home
last modified time | relevance | path

Searched defs:tensor_data (Results 1 – 25 of 43) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconvert_to_constants.py100 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
187 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
283 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
368 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
381 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
393 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
413 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
448 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
463 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
527 def convert_variable_to_constant(self, incoming_edge, tensor_data): argument
[all …]
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DStaging.cpp115 const ValueRef tensor_data, in add_prepack_standard_node()
146 const ValueRef tensor_data, in prepack_standard()
161 const ValueRef tensor_data, in prepack_standard_like()
175 const ValueRef tensor_data, in add_prepack_direct_copy_buffer_node()
200 const ValueRef tensor_data) { in prepack_direct_copy_buffer()
/aosp_15_r20/external/federated-compute/fcp/aggregation/core/
H A Dtensor_data_test.cc47 MockTensorData tensor_data(0, 0); in TEST() local
52 MockTensorData tensor_data(0, 33); in TEST() local
57 MockTensorData tensor_data(3, 100); in TEST() local
62 MockTensorData tensor_data(0, 96); in TEST() local
H A Dtensor.cc210 auto tensor_data = std::make_unique<SerializedContentStringData>(); in DecodeContent() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor.cpp200 std::optional<at::Tensor> tensor_data = CurrentTensorData(); in GetIrValue() local
210 void LazyTensor::SetTensorData(at::Tensor tensor_data) { in SetTensorData()
239 std::optional<at::Tensor> tensor_data = CurrentTensorData(); in ToTensor() local
H A Dtensor.h50 std::optional<at::Tensor> tensor_data; member
H A Dlazy_graph_executor.cpp614 std::optional<at::Tensor> tensor_data = tensors[i]->CurrentTensorData(); in CollectSyncTensors() local
1000 std::optional<at::Tensor> tensor_data = in FetchTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_add_op.cc292 const T* tensor_data, float min_tensor, in VectorTensorAddition()
310 const quint8* tensor_data, float min_tensor, in VectorTensorAddition()
395 const quint8* tensor_data, float min_tensor, in VectorTensorAddition()
537 const T* tensor_data; in Compute() local
H A Dquantized_mul_op.cc191 int64_t vector_num_elements, const T* tensor_data, in VectorTensorMultiply()
205 const quint8* tensor_data, int32 tensor_offset, int64 tensor_num_elements, in VectorTensorMultiply()
345 const T* tensor_data; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfill_test.cc81 const TensorData& tensor_data, float value) { in QuantizedFillOpModel()
189 const TensorData tensor_data(GetTensorType<quant_type>(), {}, in QuantizedFill() local
H A Dspace_to_depth_test.cc33 SpaceToDepthOpModel(const TensorData& tensor_data, int block_size) { in SpaceToDepthOpModel()
H A Ddepth_to_space_test.cc34 DepthToSpaceOpModel(const TensorData& tensor_data, int block_size) { in DepthToSpaceOpModel()
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_io.cpp237 float* tensor_data = tensor.data(); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
285 float* tensor_data = tensor.data(); in test_block_io_copy_using_reordered_dimensions_squeeze() local
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dtensor_crc32.cc27 StringPiece tensor_data = tensor.tensor_data(); in TensorToCRC32() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dquantized_test.cpp159 auto* tensor_data = tensor.mutable_data_ptr<float>(); in TEST() local
191 auto* tensor_data = tensor.data_ptr<float>(); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder_helper.h103 absl::Status CreateVectorCopyData(const TfLiteTensor& tensor, T* tensor_data) { in CreateVectorCopyData()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dspace_to_depth_test.cc23 SpaceToDepthOpModel(const TensorData& tensor_data, int block_size, in SpaceToDepthOpModel()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmemmapped_file_system_writer.cc43 const auto tensor_data = tensor.tensor_data(); in SaveTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dtflite_subgraph_execute.cc193 tensorflow::StringPiece tensor_data = tf_tensor.tensor_data(); in SetSubgraphInput() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DTensor.cpp104 TensorBase TensorBase::tensor_data() const { in tensor_data() function in at::TensorBase
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_tensor.cc120 void* tensor_data = tensor_interface->Data(); in Print() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dstring_util_test.cc47 } tensor_data; in TEST() member
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DTensorBody.h560 at::Tensor tensor_data() const { in tensor_data() function
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dplanner.py59 tensor_data: Optional[TensorWriteData] = None variable in WriteItem
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dsnapshot_utils_test.cc33 std::string tensor_data(1024, 'a'); in GenerateTensorVector() local

12