Searched refs:default_value_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | hashtable_find.cc | 43 const TfLiteTensor* default_value_tensor; in PrepareHashtableFind() local
45 &default_value_tensor)); in PrepareHashtableFind()
53 TF_LITE_ENSURE_EQ(context, default_value_tensor->type, output_tensor->type); in PrepareHashtableFind()
71 const TfLiteTensor* default_value_tensor; in EvalHashtableFind() local
73 &default_value_tensor)); in EvalHashtableFind()
84 lookup->Lookup(context, key_tensor, output_tensor, default_value_tensor); in EvalHashtableFind()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | ragged_tensor_to_tensor_op.cc | 483 const auto& default_value_tensor = context->input(kDefaultValueInputIndex); in SetOutput() local
495 default_value_tensor.flat<VALUE_TYPE>().data(); in SetOutput()
497 if (default_value_tensor.NumElements() != value_element_size && in SetOutput()
498 default_value_tensor.NumElements() != 1) { in SetOutput()
499 const auto& src_shape = default_value_tensor.shape(); in SetOutput()
507 context->allocate_temp(default_value_tensor.dtype(), in SetOutput()
511 device, context, bcast_default, element_shape, default_value_tensor, in SetOutput()
552 if (default_value_tensor.NumElements() == 1) { in SetOutput()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| H A D | ragged_tensor_to_tensor_tflite.cc | 453 const TfLiteTensor& default_value_tensor, in SetOutputT() argument
458 GetTensorData<VALUE_TYPE>(&default_value_tensor); in SetOutputT()
528 const TfLiteTensor& default_value_tensor, in SetOutput() argument
533 default_value_tensor, output_tensor); in SetOutput()
537 default_value_tensor, output_tensor); in SetOutput()
541 default_value_tensor, output_tensor); in SetOutput()
|