| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_test.cc | 316 template <TensorType tensor_type, typename integer_dtype>
326 QuantizedAddOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
327 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
328 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsNoActivation()
374 template <enum TensorType tensor_type, typename integer_dtype>
384 QuantizedAddOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
385 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
386 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
406 template <enum TensorType tensor_type, typename integer_dtype>
412 QuantizedAddOpModel m({tensor_type, test_shapes[i], -3.0, 3.0}, in QuantizedVariousInputShapes()
[all …]
|
| D | div_test.cc | 210 template <TensorType tensor_type, typename integer_dtype>
213 QuantizedDivOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedNoActivation()
214 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedNoActivation()
215 {tensor_type, {}, -1.0, 1.0}, in QuantizedNoActivation()
229 template <TensorType tensor_type, typename integer_dtype>
239 QuantizedDivOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedActivationRELU_N1_TO_1()
240 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedActivationRELU_N1_TO_1()
241 {tensor_type, {}, -1.0, 1.0}, in QuantizedActivationRELU_N1_TO_1()
257 template <TensorType tensor_type, typename integer_dtype>
263 QuantizedDivOpModel m({tensor_type, test_shapes[i], -3.0, 3.0}, in QuantizedVariousInputShapes()
[all …]
|
| D | mul_test.cc | 292 template <TensorType tensor_type, typename integer_dtype>
294 QuantizedMulOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in NoActivation()
295 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in NoActivation()
296 {tensor_type, {}, -1.0, 1.0}, in NoActivation()
306 template <TensorType tensor_type, typename integer_dtype>
310 QuantizedMulOpModel m({tensor_type, {1, 2, 2, 1}, -100, 100}, in NoActivationLargeMultiplier()
311 {tensor_type, {1, 2, 2, 1}, -100, 100}, in NoActivationLargeMultiplier()
312 {tensor_type, {}, -10, 10}, in NoActivationLargeMultiplier()
369 template <TensorType tensor_type, typename integer_dtype>
377 {tensor_type, {}, kMinUint8, kMaxUint8}, in NoActivationInt16With8BitOutput()
[all …]
|
| D | sub_test.cc | 313 template <TensorType tensor_type, typename integer_dtype>
324 QuantizedSubOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
325 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
326 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsNoActivation()
350 template <TensorType tensor_type, typename integer_dtype>
360 QuantizedSubOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
361 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
362 {tensor_type, {}, -1.0, 1.0}, in QuantizedTestsActivationRELU_N1_TO_1()
385 template <TensorType tensor_type, typename integer_dtype>
391 QuantizedSubOpModel m({tensor_type, test_shapes[i], -3.0, 3.0}, in QuantizedVariousInputShapes()
[all …]
|
| D | reduce_test.cc | 760 template <TensorType tensor_type, typename integer_dtype>
762 const float input_min = (tensor_type == TensorType_INT16) ? -24.0 : 0.0; in ConstIntProdOpTestNotKeepDimsLarge()
765 (tensor_type == TensorType_INT16) ? -1.9619905536e+12 : 0.0; in ConstIntProdOpTestNotKeepDimsLarge()
771 ProdOpConstModel m({tensor_type, {4, 3, 2}, input_min, input_max}, in ConstIntProdOpTestNotKeepDimsLarge()
772 {tensor_type, {2}, output_min, output_max}, {4}, in ConstIntProdOpTestNotKeepDimsLarge()
808 template <TensorType tensor_type, typename integer_dtype>
813 ProdOpConstModel m({tensor_type, {4, 3, 2}, -1.3, 1.3}, in ConstIntProdOpTestNotKeepDimsSmall()
814 {tensor_type, {2}, -0.0062270208, 0.0062270208}, {4}, in ConstIntProdOpTestNotKeepDimsSmall()
905 template <TensorType tensor_type, typename integer_dtype>
907 const float input_min = (tensor_type == TensorType_INT16) ? -24.0 : 0.0; in DynamicIntProdOpTestKeepDims()
[all …]
|
| D | l2norm_test.cc | 34 const TensorType tensor_type, in L2NormOpModel() argument
36 TensorData data = TensorData{tensor_type}; in L2NormOpModel()
37 if (tensor_type != TensorType_FLOAT32) { in L2NormOpModel()
44 if (tensor_type != TensorType_FLOAT32) { in L2NormOpModel()
|
| D | basic_rnn_test.cc | 235 HybridRNNOpModel(int batches, int units, int size, TensorType tensor_type, in HybridRNNOpModel() argument
237 : RNNOpModel(batches, units, size, tensor_type, tensor_type, in HybridRNNOpModel()
239 tensor_type_ = tensor_type; in HybridRNNOpModel()
|
| D | svdf_test.cc | 207 int rank, TensorType tensor_type, in HybridSVDFOpModel() argument
210 tensor_type, tensor_type, asymmetric_quantize_inputs) { in HybridSVDFOpModel()
211 tensor_type_ = tensor_type; in HybridSVDFOpModel()
|
| D | unidirectional_sequence_rnn_test.cc | 253 TensorType tensor_type, in HybridUnidirectionalRNNOpModel() argument
256 tensor_type, tensor_type, in HybridUnidirectionalRNNOpModel()
258 tensor_type_ = tensor_type; in HybridUnidirectionalRNNOpModel()
|
| D | matrix_set_diag_test.cc | 78 {TypeParam::tensor_type, {3, 3}}, {TypeParam::tensor_type, {3}}); in TYPED_TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | mul_test.cc | 59 template <TensorType tensor_type, typename integer_dtype>
62 /*input1=*/{tensor_type, {2, 3}, -0.44f, 8.0f}, in TestMulOutputImpl()
63 /*input2=*/{tensor_type, {1, 3}, 0, 0.999f}, in TestMulOutputImpl()
64 /*output=*/{tensor_type, {2, 3}, -1.0f, 1.0f}, activation_func); in TestMulOutputImpl()
78 template <TensorType tensor_type, typename integer_dtype>
81 /*input1=*/{tensor_type, {1, 2, 2, 3}, -0.44f, 55.7f}, in TestLargeInputRangeImpl()
82 /*input2=*/{tensor_type, {1, 1, 2, 3}, 0, 0.999f}, in TestLargeInputRangeImpl()
83 /*output=*/{tensor_type, {1, 2, 2, 3}, -1.0f, 1.0f}, activation_func); in TestLargeInputRangeImpl()
|
| D | arithmetic_test.cc | 99 template <TensorType tensor_type, typename integer_dtype>
107 AddOpModel m({tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
108 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, in QuantizedTestsNoActivation()
109 {tensor_type, {1, 2, 2, 1}, -1.0, 1.0}, activation_func); in QuantizedTestsNoActivation()
|
| D | l2_norm_test.cc | 24 const TensorType tensor_type) { in L2NormOpModel() argument
25 TensorData data = TensorData{tensor_type}; in L2NormOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | constant_fold.cc | 53 auto tensor_type = t.cast<TensorType>(); in ShouldBeFolded() local
55 if (!tensor_type.getElementType().isIntOrFloat()) continue; in ShouldBeFolded()
56 if (!tensor_type.hasStaticShape()) { in ShouldBeFolded()
60 size += tensor_type.getNumElements() * in ShouldBeFolded()
61 tensor_type.getElementType().getIntOrFloatBitWidth(); in ShouldBeFolded()
94 if (auto tensor_type = type.dyn_cast<TensorType>()) { in ConstantFoldFallbackHook() local
95 if (tensor_type.getElementType().isa<VariantType>()) { in ConstantFoldFallbackHook()
|
| D | readonly_references_to_resources.cc | 160 TensorType tensor_type = DropRefType(shaped_type).cast<TensorType>(); in runOnFunction() local
172 {}, TF::ResourceType::get(ArrayRef<TensorType>{tensor_type}, in runOnFunction()
184 user->getLoc(), ArrayRef<Type>{tensor_type}, in runOnFunction()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Recv.pbtxt | 5 type_attr: "tensor_type"
8 name: "tensor_type"
40 type_attr: "tensor_type"
43 name: "tensor_type"
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier_test.cc | 578 for (int tensor_type = TensorType_MIN; tensor_type <= TensorType_MAX; in TEST() local
579 ++tensor_type) { in TEST()
580 if (tensor_type == TensorType_STRING) continue; in TEST()
581 builder.AddTensor({2, 3}, static_cast<TensorType>(tensor_type), in TEST()
595 for (int tensor_type = TensorType_MIN; tensor_type <= TensorType_MAX; in TEST() local
596 ++tensor_type) { in TEST()
597 if (tensor_type == TensorType_STRING || in TEST()
598 tensor_type == TensorType_RESOURCE || tensor_type == TensorType_VARIANT) in TEST()
601 ASSERT_EQ(ConvertTensorType(static_cast<TensorType>(tensor_type), in TEST()
608 builder.AddTensor({1}, static_cast<TensorType>(tensor_type), buffer, in TEST()
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_sycl.cpp | 99 using tensor_type = Tensor<DataType, 1, DataLayout, IndexType>; in test_sycl_mem_sync_offsets() typedef
103 tensor_type in1(tensorRange); in test_sycl_mem_sync_offsets()
104 tensor_type out(tensorRange); in test_sycl_mem_sync_offsets()
107 TensorMap<tensor_type> gpu1(gpu_data, tensorRange); in test_sycl_mem_sync_offsets()
135 TensorMap<tensor_type> gpu2(gpu_data_out, tensorRange); in test_sycl_mem_sync_offsets()
153 using tensor_type = Tensor<DataType, 1, DataLayout, IndexType>; in test_sycl_memset_offsets() typedef
157 tensor_type cpu_out(tensorRange); in test_sycl_memset_offsets()
158 tensor_type out(tensorRange); in test_sycl_memset_offsets()
166 TensorMap<tensor_type> gpu1(gpu_data, tensorRange); in test_sycl_memset_offsets()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Recv.pbtxt | 5 type_attr: "tensor_type"
8 name: "tensor_type"
|
| /external/XNNPACK/test/ |
| D | subgraph-tester.h | 45 xnn_tensor_type tensor_type, in add_tensor() argument
48 if (tensor_type == kStaticDense || tensor_type == kStaticSparse) { in add_tensor()
53 if (tensor_type == kStaticDense) { in add_tensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 137 auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>(); in CastToNonDerivedType() local
138 if (!tensor_type) return cast_op.arg(); in CastToNonDerivedType()
140 auto attr_names = tensor_type.getAttrKeys(); in CastToNonDerivedType()
190 if (auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>()) { in AddDerivedAttrs() local
191 auto attr_names = tensor_type.getAttrKeys(); in AddDerivedAttrs()
331 if (auto tensor_type = res.value().dyn_cast<TFRTensorType>()) { in DeriveOutputTypes() local
333 auto attr_key = tensor_type.getAttrKeys().front(); in DeriveOutputTypes()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_common.cc | 41 mlir::Type tensor_type, in ParseExecuteOpCommon() argument
97 parser.resolveOperands(operands, tensor_type, result.operands)) in ParseExecuteOpCommon()
101 result.types.append(num_results, tensor_type); in ParseExecuteOpCommon()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 141 DataType tensor_type; in Compute() local
144 old_tensor_name, &tensor_type, &tensor_shape)); in Compute()
145 OP_REQUIRES(context, tensor_type == DT_FLOAT, in Compute()
148 DataTypeString(tensor_type), " instead of expected type ", in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | default_quant_params.cc | 160 auto tensor_type = value.getType().dyn_cast<TensorType>(); in AddToWorkListIfUnquantized() local
161 if (!tensor_type) { in AddToWorkListIfUnquantized()
165 if (!tensor_type.getElementType().isF32()) return; in AddToWorkListIfUnquantized()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | keras_tensor.py | 589 for tensor_type, cls in keras_tensor_classes:
590 if isinstance(tensor, tensor_type):
607 for tensor_type, cls in keras_tensor_classes:
608 if issubclass(value_type, tensor_type):
|