Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_helpers.inc483 auto rhs_attr = operands[1].dyn_cast_or_null<DenseElementsAttr>();
484 if (!rhs_attr && !(is_commutative && lhs_attr)) return {};
505 if (rhs_attr && is_valid_broadcasting(lhs_type, rhs_type, result_type)) {
506 if (rhs_attr.isSplat() && rhs_attr.getSplatValue() == identity_attr)
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc217 DataType rhs_attr = GetDataTypeFromAttr(*rhs, type_attr); in HaveSameDataType() local
219 return lhs_attr != DT_INVALID && rhs_attr != DT_INVALID && in HaveSameDataType()
220 lhs_attr == rhs_attr; in HaveSameDataType()