Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_import.cc47 const TfLiteTensor* value_tensor; in PrepareHashtableImport() local
69 const TfLiteTensor* value_tensor; in EvalHashtableImport() local
Dhashtable_ops_test.cc738 TfLiteTensor value_tensor = CreateTensor<ValueType>(value_type, values); in InitHashtableResource() local
769 auto value_tensor = interpreter_->tensor(values_); in CreateHashtableResource() local
/external/tensorflow/tensorflow/python/keras/layers/
Dmulti_head_attention_test.py253 value_tensor, argument
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dlmdb_dataset_op.cc98 Tensor& value_tensor = out_tensors->back(); in GetNextInternal() local
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.cc338 Tensor value_tensor(input.dtype(), shape_1d); in GrapplerItemFromMetaGraphDef() local
/external/tensorflow/tensorflow/core/kernels/
Dbias_op.cc45 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc2745 const auto& value_tensor = context->tensors[node->inputs->data[1]]; in Validate() local
4615 const TfLiteTensor& value_tensor = context->tensors[value_id]; in AddOpsAndTensors() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc3722 Tensor value_tensor(DT_FLOAT, TensorShape({})); in TEST_F() local