cast_op.cc | 30 OP_REQUIRES_OK(ctx, ctx->GetAttr("DstT", &dst_dtype_)); 32 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dst_dtype_, &dst_type_)); 40 if (src_dtype_ == dst_dtype_) { 42 } else if (dst_dtype_ == DT_BOOL) { 57 DataType src_dtype_, dst_dtype_; member in class:tensorflow::__anon38730::CastOp
|