Home
last modified time | relevance | path

Searched defs:tensor_type (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_fold.cc53 auto tensor_type = t.cast<TensorType>(); in ShouldBeFolded() local
94 if (auto tensor_type = type.dyn_cast<TensorType>()) { in ConstantFoldFallbackHook() local
Dreadonly_references_to_resources.cc160 TensorType tensor_type = DropRefType(shaped_type).cast<TensorType>(); in runOnFunction() local
/external/tflite-support/tensorflow_lite_support/cc/task/vision/utils/
Dimage_tensor_specs.h67 TfLiteType tensor_type; member
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc137 auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>(); in CastToNonDerivedType() local
190 if (auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>()) { in AddDerivedAttrs() local
331 if (auto tensor_type = res.value().dyn_cast<TFRTensorType>()) { in DeriveOutputTypes() local
Ddecompose.cc180 if (auto tensor_type = arg.value().dyn_cast<TFRTensorType>()) { in RewriteUnregisteredTFOps() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dl2_norm_test.cc24 const TensorType tensor_type) { in L2NormOpModel()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor.cc256 if (auto tensor_type = type.dyn_cast<mlir::RankedTensorType>()) { in ConvertTypeToTensorShape() local
272 if (auto tensor_type = type.dyn_cast<mlir::RankedTensorType>()) { in ConvertTypeToTensorShapeAttr() local
Dcompile_mlir_util.cc169 auto tensor_type = type_and_idx.value().dyn_cast<mlir::RankedTensorType>(); in GetOutputInfo() local
576 auto tensor_type = mlir_arg.getType().cast<mlir::TensorType>(); in RewriteWithArgs() local
/external/eigen/unsupported/test/
Dcxx11_tensor_sycl.cpp99 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/
Dcompatibility_analysis.cc143 if (auto tensor_type = type.dyn_cast<mlir::TensorType>()) { in AnalyzeVariable() local
/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
Dtfrt_fallback_common.cc41 mlir::Type tensor_type, in ParseExecuteOpCommon()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_value_typed_analyzer.cc29 if (auto tensor_type = type.dyn_cast<TensorType>()) { in IsResourceType() local
/external/tensorflow/tensorflow/lite/tools/
Dvisualize.py209 def TensorTypeToName(tensor_type): argument
Dverifier_test.cc578 for (int tensor_type = TensorType_MIN; tensor_type <= TensorType_MAX; in TEST() local
595 for (int tensor_type = TensorType_MIN; tensor_type <= TensorType_MAX; in TEST() local
/external/tensorflow/tensorflow/lite/kernels/
Dl2norm_test.cc34 const TensorType tensor_type, in L2NormOpModel()
Dbasic_rnn_test.cc235 HybridRNNOpModel(int batches, int units, int size, TensorType tensor_type, in HybridRNNOpModel()
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc141 DataType tensor_type; in Compute() local
/external/tensorflow/tensorflow/lite/toco/tflite/
Dtypes.cc114 ArrayDataType DataType::Deserialize(int tensor_type) { in Deserialize()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dutils.cc33 TfLiteType tensor_type) { in TensorTypeMatch()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddefault_quant_params.cc160 auto tensor_type = value.getType().dyn_cast<TensorType>(); in AddToWorkListIfUnquantized() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.cc470 ElementsAttr QuantizeLegacy(Attribute real_value, Type tensor_type) { in QuantizeLegacy()
563 ElementsAttr Quantize(Attribute real_value, Type tensor_type) { in Quantize()
793 Type tensor_type, double scale, in GetFixedOutputRange()
/external/XNNPACK/test/
Dsubgraph-tester.h45 xnn_tensor_type tensor_type, in add_tensor()
/external/tensorflow/tensorflow/lite/tools/strip_buffers/
Dstripping_lib.cc81 TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type) { in ConvertTensorType()
/external/tensorflow/tensorflow/python/ops/
Dio_ops.py77 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type, argument
/external/tensorflow/tensorflow/python/keras/layers/
Dadvanced_activations.py277 def _large_compatible_negative(tensor_type): argument

12