Searched refs:SetDataTypeToAttr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 130 void SetDataTypeToAttr(DataType dtype, const string& attr_name, NodeDef* node) { in SetDataTypeToAttr() function
1266 SetDataTypeToAttr(operand_input_type, "T", bitcast); in TrySimplify()
2019 SetDataTypeToAttr(dtype, type_attr_name, node); in SetInputType()
2446 SetDataTypeToAttr(type, "T", new_mul_node); in TrySimplify()
3275 SetDataTypeToAttr(output_properties.dtype(), "T", output); in RewriteGraph()
3283 SetDataTypeToAttr(DT_INT32, "dtype", axis); in RewriteGraph()
3290 SetDataTypeToAttr(output_properties.dtype(), "T", output); in RewriteGraph()
|