Home
last modified time | relevance | path

Searched refs:input_defining_op (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_ops_for_outside_compilation.cc241 Operation* input_defining_op = value.getDefiningOp(); in MarkVariantInputsOutputs() local
242 if (IsVariant(value) && input_defining_op && in MarkVariantInputsOutputs()
243 !HasOutsideCompiledAncestor(input_defining_op) && in MarkVariantInputsOutputs()
244 !input_defining_op->hasAttrOfType<StringAttr>( in MarkVariantInputsOutputs()
246 input_defining_op->setAttr( in MarkVariantInputsOutputs()
248 StringAttr::get(input_defining_op->getContext(), "auto")); in MarkVariantInputsOutputs()
249 outside_compiled_ops.push(input_defining_op); in MarkVariantInputsOutputs()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc417 auto input_defining_op = dyn_cast<TFL::QConstOp>(input.getDefiningOp()); in getUnquantizedBias() local
418 auto dense_attr = input_defining_op.value().dyn_cast<DenseElementsAttr>(); in getUnquantizedBias()