Home
last modified time | relevance | path

Searched defs: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
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc579 TensorType &output_ty) { in GetReshapeOutputType()
677 auto output_ty = op.getType().dyn_cast<RankedTensorType>(); in Verify() local
705 TensorType output_ty; in build() local
1090 auto output_ty = op.output().getType().dyn_cast<RankedTensorType>(); in Verify() local
1988 auto output_ty = output().getType().dyn_cast<RankedTensorType>(); in fold() local
Dtf_ops_a_m.cc213 RankedTensorType output_ty = GetRankedTensorTypeForOperand(op.output()); in Verify() local
/external/tensorflow/tensorflow/lite/python/
Dlite.py322 def quantizer_flags(self, input_ty=None, output_ty=None): argument
347 def flags_modify_model_io_type(self, input_ty=None, output_ty=None): argument
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1351 TensorType &output_ty) { in GetReshapeOutputType()
1451 auto output_ty = op.getType().dyn_cast<RankedTensorType>(); in Verify() local