Home
last modified time | relevance | path

Searched defs:value_tensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dhashtable_import.cc45 const TfLiteTensor* value_tensor; in PrepareHashtableImport() local
67 const TfLiteTensor* value_tensor; in EvalHashtableImport() local
H A Dhashtable_ops_test.cc725 TfLiteTensor value_tensor = CreateTensor<ValueType>(value_type, values); in InitHashtableResource() local
755 auto value_tensor = interpreter_->tensor(values_); in CreateHashtableResource() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfilesystem_ops.cc43 const Tensor* value_tensor; in Compute() local
H A Dbias_op.cc48 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Dsmall_constant_optimization.cc96 std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> value_tensor( in ExtractSmallTensorValue() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dlmdb_dataset_op.cc98 Tensor& value_tensor = out_tensors->back(); in GetNextInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgrappler_item_builder.cc339 Tensor value_tensor(input.dtype(), shape_1d); in GrapplerItemFromMetaGraphDef() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc3431 const auto& value_tensor = context->tensors[node->inputs->data[1]]; in Validate() local
6014 const TfLiteTensor& value_tensor = context->tensors[value_id]; in AddOpsAndTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding_test.cc3727 Tensor value_tensor(DT_FLOAT, TensorShape({})); in TEST_F() local