Home
last modified time | relevance | path

Searched defs:tensor_value (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/shape_inference/
H A Dpass.cc112 DenseElementsAttr tensor_value; in TryToCacheResultsTensorValue() local
125 DenseElementsAttr tensor_value; in TryToCacheResultsTensorValue() local
146 DenseElementsAttr tensor_value; in TryToCacheResultsTensorValue() local
/aosp_15_r20/external/executorch/runtime/executor/
H A Dmethod_meta.cpp135 auto tensor_value = s_plan_->values()->Get(input_index)->val_as_Tensor(); in input_tensor_meta() local
176 auto tensor_value = s_plan_->values()->Get(output_index)->val_as_Tensor(); in output_tensor_meta() local
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddebug_events_monitors_test.py347 self, tensor_value, dtype, expected_size, expected_num_neg_inf, argument
440 self, tensor_value, dtype, expected_size, expected_num_neg_inf, argument
H A Ddebug_events_monitors.py148 def _check_full_tensor_value(self, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtensor_array_ops.cc422 const Tensor* tensor_value; in Compute() local
1031 const Tensor* tensor_value; in Compute() local
1226 const Tensor* tensor_value; in Compute() local
H A Dquantized_add_op.cc375 const int64 tensor_value = static_cast<int64_t>(tensor_data[i]); in VectorTensorAddition() local
435 const int64_t tensor_value = static_cast<int64_t>(tensor_data[i]); in VectorTensorAddition() local
/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_block_io.cpp257 float tensor_value = tensor_tensor(d0, d1, d2); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
306 float tensor_value = tensor_tensor(d0, d1, d2, d3); in test_block_io_copy_using_reordered_dimensions_squeeze() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DFillBorder.cpp67 const uint8_t tensor_value = distribution_u8(generator); variable
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/
H A DFillBorder.cpp66 const uint8_t tensor_value = distribution_u8(generator); variable
/aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/compiler/
H A Dxnn_compiler.cpp46 auto tensor_value = value->xvalue_as_XNNTensorValue(); in compileModel() local
/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/
H A Dexecution_trace_observer.cpp360 std::string tensor_shape, tensor_stride, tensor_type, tensor_value; in convertIValue() local
454 std::string tensor_value = getScalarValue(val); in convertIValue() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dop_kernel_runner.h113 for (const auto& tensor_value : tensor_values) { in OpKernelRunState() local
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Doptimizer.py289 def wrap_tensor(self, tx: "InstructionTranslator", tensor_value): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/
H A Dtfprof_output.proto22 TFProfTensorProto tensor_value = 15; field
/aosp_15_r20/external/executorch/backends/xnnpack/runtime/
H A DXNNCompiler.cpp163 const fb_xnnpack::XNNTensorValue* tensor_value, in getConstantDataPtr()
198 const fb_xnnpack::XNNTensorValue* tensor_value = nullptr; in defineTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Ddebug_events_writer.cc278 const Tensor& tensor_value) { in WriteGraphExecutionTrace()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dkernel_and_device.cc72 for (const TensorValue& tensor_value : tensor_args_) { in GetLocalTensors() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dmemory_planner.cpp122 for (auto* tensor_value : new_free_tensors) { in assignStorageToManagedTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/benchmark/
H A Dbenchmark_test.cc118 StringRef tensor_value = GetString(input_tensor, tensor_dim_index); in CheckInputTensorValue() local
/aosp_15_r20/external/federated-compute/fcp/aggregation/protocol/simple_aggregation/
H A Dsimple_aggregation_protocol_test.cc814 std::atomic<int> tensor_value = 0; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc1509 const std::vector<T>& tensor_value, in AddNewInputConstantTensor()
1561 const std::vector<T>& tensor_value, in AddNewInputConstantTensor()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dshape_type_inference.cpp427 auto tensor_value = in ComputeConstantFolding() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter_test.cc1098 int* tensor_value = interpreter.typed_tensor<int>(1); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc114 ValueRange control_operands, TypedAttr tensor_value, in CreateConstantTensorOp()