| /external/tensorflow/tensorflow/lite/kernels/ |
| D | hashtable_import.cc | 45 const TfLiteTensor* value_tensor; in PrepareHashtableImport() local
47 GetInputSafe(context, node, kValueTensor, &value_tensor)); in PrepareHashtableImport()
49 value_tensor->type == kTfLiteString) || in PrepareHashtableImport()
51 value_tensor->type == kTfLiteInt64)); in PrepareHashtableImport()
54 TF_LITE_ENSURE(context, HaveSameShapes(key_tensor, value_tensor)); in PrepareHashtableImport()
67 const TfLiteTensor* value_tensor; in EvalHashtableImport() local
69 GetInputSafe(context, node, kValueTensor, &value_tensor)); in EvalHashtableImport()
76 lookup->CheckKeyAndValueTypes(context, key_tensor, value_tensor)); in EvalHashtableImport()
79 auto result = lookup->Import(context, key_tensor, value_tensor); in EvalHashtableImport()
|
| D | hashtable_ops_test.cc | 723 TfLiteTensor value_tensor = CreateTensor<ValueType>(value_type, values); in InitHashtableResource() local
724 lookup->Import(&context, &key_tensor, &value_tensor); in InitHashtableResource()
726 TfLiteTensorFree(&value_tensor); in InitHashtableResource()
753 auto value_tensor = interpreter_->tensor(values_); in CreateHashtableResource() local
757 &resources, resource_id, key_tensor->type, value_tensor->type); in CreateHashtableResource()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 47 void GetBiasValueDims(const Tensor& value_tensor, TensorFormat data_format, in GetBiasValueDims() argument
56 int32_t channel_dim = value_tensor.dims() - 1; in GetBiasValueDims()
57 *channel = static_cast<int32>(value_tensor.dim_size(channel_dim)); in GetBiasValueDims()
59 *batch *= static_cast<int32>(value_tensor.dim_size(i)); in GetBiasValueDims()
62 *batch = static_cast<int32>(value_tensor.dim_size(0)); in GetBiasValueDims()
63 *channel = static_cast<int32>(value_tensor.dim_size(1)); in GetBiasValueDims()
64 *height = static_cast<int32>(value_tensor.dim_size(2)); in GetBiasValueDims()
65 if (value_tensor.dims() > 3) { in GetBiasValueDims()
66 *width = static_cast<int32>(value_tensor.dim_size(3)); in GetBiasValueDims()
68 if (value_tensor.dims() > 4) { in GetBiasValueDims()
[all …]
|
| /external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/ |
| D | static_hashtable.py | 43 value_tensor = tf.constant(values, dtype=value_dtype)
45 initializer = tf.lookup.KeyValueTensorInitializer(key_tensor, value_tensor)
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention_test.py | 214 value_tensor = keras.Input(value_shape[1:], name="value")
216 output = test_layer(query=query_tensor, value=value_tensor,
218 model = keras.Model([query_tensor, value_tensor, mask_tensor], output)
249 value_tensor, argument
252 return value_tensor, None
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | lmdb_dataset_op.cc | 98 Tensor& value_tensor = out_tensors->back(); in GetNextInternal() local
99 value_tensor.scalar<tstring>()() = string( in GetNextInternal()
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure_test.py | 663 value_tensor = constant_op.constant(42.0)
664 s_tensor = structure.type_spec_from_value(value_tensor)
665 flat_tensor = structure.to_tensor_list(s_tensor, value_tensor)
690 structure.to_tensor_list(s_sparse_tensor, value_tensor)
698 structure.to_tensor_list(s_nest, value_tensor)
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 338 Tensor value_tensor(input.dtype(), shape_1d); in GrapplerItemFromMetaGraphDef() local
339 InitializeTensor(input.dtype(), &value_tensor); in GrapplerItemFromMetaGraphDef()
341 NodeName(input.coo_sparse().values_tensor_name()), value_tensor); in GrapplerItemFromMetaGraphDef()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 308 value_tensor = ops.convert_to_tensor(value)
309 shape.assert_is_compatible_with(value_tensor.shape)
311 handle, value_tensor, name=name)
891 value_tensor = ops.convert_to_tensor(value, dtype=self.dtype)
892 if not self._shape.is_compatible_with(value_tensor.shape):
902 self.handle, value_tensor, name=name)
|
| D | metrics_impl.py | 1608 value_tensor = _aggregate_variable(count, metrics_collections)
1614 return value_tensor, update_op
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 2964 const auto& value_tensor = context->tensors[node->inputs->data[1]]; in Validate() local
2965 EXPECT_INPUT_TYPE_IN(value_tensor.type, kTfLiteFloat32, kTfLiteInt32, in Validate()
2967 if (value_tensor.type == kTfLiteInt64 && in Validate()
2968 IsConstantTensor(&value_tensor)) { in Validate()
2970 *value_tensor.data.i64 <= std::numeric_limits<int32_t>::max() && in Validate()
2971 *value_tensor.data.i64 >= std::numeric_limits<int32_t>::min(), in Validate()
5137 const TfLiteTensor& value_tensor = context->tensors[value_id]; in AddOpsAndTensors() local
5138 switch (value_tensor.type) { in AddOpsAndTensors()
5140 if (value_tensor.allocation_type == kTfLiteMmapRo) { in AddOpsAndTensors()
5142 builder.AddScalarFloat32Operand(*value_tensor.data.f)); in AddOpsAndTensors()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding_test.cc | 3727 Tensor value_tensor(DT_FLOAT, TensorShape({})); in TEST_F() local
3728 value_tensor.flat<float>()(0) = 1; in TEST_F()
3729 value_tensor.AsProtoTensorContent(value.mutable_tensor()); in TEST_F()
|