Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_fold.cc50 auto tensor_type = t.cast<TensorType>(); in ShouldBeFolded() local
84 if (auto tensor_type = type.dyn_cast<TensorType>()) { in ConstantFoldFallbackHook() local
Dreadonly_references_to_resources.cc140 TensorType tensor_type = DropRefType(shaped_type).cast<TensorType>(); in runOnFunction() local
Doptimize_global_tensors.cc59 if (auto tensor_type = type.dyn_cast<TensorType>()) { in IsResourceType() local
/external/tensorflow/tensorflow/lite/tools/versioning/
Dop_version.cc839 TensorType tensor_type = GetTensorType(op->inputs()->Get(i), subgraph); in GetOpSignature() local
843 TensorType tensor_type = GetTensorType(op->outputs()->Get(i), subgraph); in GetOpSignature() 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.cc127 auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>(); in CastToNonDerivedType() local
188 if (auto tensor_type = input_tfr_type.dyn_cast<TFRTensorType>()) { in AddDerivedAttrs() local
328 if (auto tensor_type = res.value().dyn_cast<TFRTensorType>()) { in DeriveOutputTypes() local
Ddecompose.cc158 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.cc192 if (auto tensor_type = type.dyn_cast<mlir::RankedTensorType>()) { in ConvertTypeToTensorShape() local
208 if (auto tensor_type = type.dyn_cast<mlir::RankedTensorType>()) { in ConvertTypeToTensorShapeAttr() local
Dcompile_mlir_util.cc167 auto tensor_type = type_and_idx.value().dyn_cast<mlir::RankedTensorType>(); in GetOutputInfo() local
548 auto tensor_type = mlir_arg.getType().cast<mlir::TensorType>(); in RewriteWithArgs() local
/external/tensorflow/tensorflow/lite/tools/
Dvisualize.py202 def TensorTypeToName(tensor_type): argument
Dverifier_test.cc543 for (int tensor_type = TensorType_MIN; tensor_type <= TensorType_MAX; in TEST() local
559 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()
Dunidirectional_sequence_rnn_test.cc253 TensorType tensor_type, in HybridUnidirectionalRNNOpModel()
Dsvdf_test.cc207 int rank, TensorType tensor_type, in HybridSVDFOpModel()
Dconcatenation_test.cc289 enum TensorType tensor_type = member
/external/tensorflow/tensorflow/core/kernels/
Dload_and_remap_matrix_op.cc136 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/quantization/
Dquantization_utils.cc451 ElementsAttr QuantizeLegacy(Attribute real_value, Type tensor_type) { in QuantizeLegacy()
538 ElementsAttr Quantize(Attribute real_value, Type tensor_type) { in Quantize()
750 Type tensor_type, double scale, in GetFixedOutputRange()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddefault_quant_params.cc150 auto tensor_type = value.getType().dyn_cast<TensorType>(); in AddToWorkListIfUnquantized() local
/external/XNNPACK/test/
Dsubgraph-tester.h44 xnn_tensor_type tensor_type, in add_tensor()
/external/tensorflow/tensorflow/python/ops/
Dio_ops.py75 def _restore_slice(file_pattern, tensor_name, shape_and_slice, tensor_type, argument
/external/tensorflow/tensorflow/python/keras/layers/
Dadvanced_activations.py272 def _large_compatible_negative(tensor_type): argument

12