Lines Matching defs:cast
721 Output nhwc_fp32 = ops::Cast(s, nhwc_uint8, DT_FLOAT);
747 if (node.op() == "Cast") {
757 Output nhwc_uint8 = ops::Cast(s, nhwc_fp32, DT_UINT8);
776 EXPECT_EQ(node.input(0), "Cast");
1007 // reordering cast and transpose.
1009 // Conv2D(Transpose(Mul(Cast(I), S)), W)
1011 // Conv2D(Transpose(Cast(I)), W*S)
1013 // Conv2D(Cast(Transpose(I)), W*S)
1017 Output cast = ops::Cast(s, inputs, DT_FLOAT);
1018 Output mul = ops::Mul(s, cast, ops::Const(s, 1.0f / 255.0f));
1151 Output cast = ops::Cast(s, inputs, DT_INT8);
1152 Output outputs = ops::Identity(s.WithOpName("outputs"), cast);
1164 [](const NodeDef& node) { return node.op() == "Cast"; }));
