| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | depth_to_space_tester.cc | 38 void DepthToSpaceTester::Test(TensorType tensor_type, in Test()
85 void DepthToSpaceTester::Test<float>(TensorType tensor_type, in Test()
132 void DepthToSpaceTester::Test(TensorType tensor_type, in Test()
|
| D | transpose_tester.cc | 36 void TransposeTester::Test(TensorType tensor_type, in Test()
59 void TransposeTester::Test(TensorType tensor_type, in Test()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_detensorize_linalg.cc | 50 bool IsNotZeroRankTensor(RankedTensorType tensor_type) { in IsNotZeroRankTensor()
67 auto tensor_type = in matchAndRewrite() local
|
| D | tf_jitrt_legalize_i1_type.cc | 74 addConversion([](RankedTensorType tensor_type) -> Optional<Type> { in I1TypeConverter()
84 if (auto tensor_type = type.dyn_cast<RankedTensorType>()) { in isLegalType() local
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | utils.cc | 29 auto tensor_type = operand_type.dyn_cast<TensorType>(); in HasQuantizedTensors() local
35 auto tensor_type = result_type.dyn_cast<TensorType>(); in HasQuantizedTensors() local
87 if (auto tensor_type = type.dyn_cast<TensorType>()) { in IsOperationFoldable() local
|
| D | convert_tf_quant_ops_to_mhlo.cc | 64 TensorType getSameShapeTensorType(TensorType tensor_type, Type element_type) { in getSameShapeTensorType()
|
| /external/tensorflow/tensorflow/core/ir/ |
| D | ops_test.cc | 79 ShapedType tensor_type = RankedTensorType::get(/*shape=*/{}, b.getI1Type()); in TEST() local
121 ShapedType tensor_type = RankedTensorType::get(/*shape=*/{}, b.getI32Type()); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | constant_fold.cc | 53 auto tensor_type = t.cast<TensorType>(); in ShouldBeFolded() local
99 if (auto tensor_type = type.dyn_cast<TensorType>()) { in ConstantFoldFallbackHook() local
|
| D | readonly_references_to_resources.cc | 152 TensorType tensor_type = DropRefType(shaped_type).cast<TensorType>(); in runOnOperation() local
|
| /external/tflite-support/tensorflow_lite_support/cc/task/vision/utils/ |
| D | image_tensor_specs.h | 67 TfLiteType tensor_type; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 138 auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>(); in CastToNonDerivedType() local
191 if (auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>()) { in AddDerivedAttrs() local
334 if (auto tensor_type = res.value().dyn_cast<TFRTensorType>()) { in DeriveOutputTypes() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | l2_norm_test.cc | 26 const TensorType tensor_type) { in L2NormOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | pin_ops_with_side_effects.cc | 40 const auto tensor_type = value.getType().dyn_cast<TensorType>(); in IsResourceTensor() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device_util.h | 232 virtual TensorType tensor_type() const { return TensorType::kDense; } in tensor_type() function
383 TensorType tensor_type() const override { return TensorType::kResource; } in tensor_type() function
477 TensorType tensor_type() const override { return TensorType::kSparse; } in tensor_type() function
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | concatenation_test.cc | 289 enum TensorType tensor_type = member
480 TensorType tensor_type; member
586 TensorType tensor_type = TensorType_FLOAT32; in Range() local
|
| D | l2norm_test.cc | 34 const TensorType tensor_type, in L2NormOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 249 if (auto tensor_type = type.dyn_cast<mlir::RankedTensorType>()) { in ConvertTypeToTensorShape() local
265 if (auto tensor_type = type.dyn_cast<mlir::RankedTensorType>()) { in ConvertTypeToTensorShapeAttr() local
|
| /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
153 using tensor_type = Tensor<DataType, 1, DataLayout, IndexType>; in test_sycl_memset_offsets() typedef
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/ |
| D | compatibility_analysis.cc | 143 if (auto tensor_type = type.dyn_cast<mlir::TensorType>()) { in AnalyzeVariable() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| D | tfrt_fallback_common.cc | 41 mlir::Type tensor_type, in ParseExecuteOpCommon()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_common.cc | 41 mlir::Type tensor_type, in ParseExecuteOpCommon()
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_tensor.cc | 251 if (auto tensor_type = type.dyn_cast<RankedTensorType>()) { in ConvertTypeToTensorShape() local
267 if (auto tensor_type = type.dyn_cast<RankedTensorType>()) { in ConvertTypeToTensorShapeAttr() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_value_typed_analyzer.cc | 30 if (auto tensor_type = type.dyn_cast<TensorType>()) { in IsResourceType() local
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | visualize.py | 205 def TensorTypeToName(tensor_type): argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | load_and_remap_matrix_op.cc | 146 DataType tensor_type; in Compute() local
|