Home
last modified time | relevance | path

Searched refs:output_ty (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Drewrite_tpu_embedding_ops.cc78 auto output_ty = RankedTensorType::get({}, VariantType::get(&getContext())); in runOnFunction() local
80 loc, output_ty, config); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc579 TensorType &output_ty) { in GetReshapeOutputType() argument
582 output_ty = UnrankedTensorType::get(element_ty); in GetReshapeOutputType()
597 output_ty = RankedTensorType::get(dynamic_shape, element_ty); in GetReshapeOutputType()
633 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()
662 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()
677 auto output_ty = op.getType().dyn_cast<RankedTensorType>(); in Verify() local
678 if (!output_ty) return success(); in Verify()
680 if (output_ty.hasStaticShape() && tensor_ty.hasStaticShape()) { in Verify()
681 const int64_t output_ty_size = output_ty.getNumElements(); in Verify()
689 if (!AreCastCompatible({output_ty, expected_ty})) in Verify()
[all …]
Dtf_ops_a_m.cc213 RankedTensorType output_ty = GetRankedTensorTypeForOperand(op.output()); in Verify() local
214 if (!output_ty) return success(); in Verify()
217 if (output_ty.getRank() != expected_output_rank) in Verify()
220 << " but got " << output_ty.getRank(); in Verify()
223 ArrayRef<int64_t> output_shape = output_ty.getShape(); in Verify()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1351 TensorType &output_ty) { in GetReshapeOutputType() argument
1354 output_ty = UnrankedTensorType::get(element_ty); in GetReshapeOutputType()
1369 output_ty = RankedTensorType::get(dynamic_shape, element_ty); in GetReshapeOutputType()
1408 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()
1437 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()
1451 auto output_ty = op.getType().dyn_cast<RankedTensorType>(); in Verify() local
1452 if (!output_ty) return success(); in Verify()
1454 if (output_ty.hasStaticShape() && input_ty.hasStaticShape()) { in Verify()
1455 const int64_t output_ty_size = output_ty.getNumElements(); in Verify()
1463 if (!TF::AreCastCompatible({output_ty, expected_ty})) in Verify()
[all …]
/external/tensorflow/tensorflow/lite/python/
Dlite.py322 def quantizer_flags(self, input_ty=None, output_ty=None): argument
326 inference_output_type = output_ty if output_ty else _dtypes.float32
347 def flags_modify_model_io_type(self, input_ty=None, output_ty=None): argument
353 "inference_output_type": output_ty if output_ty else _dtypes.float32,