Searched defs:cast_op (Results 1 – 11 of 11) sorted by relevance
126 CastOp cast_op, Type input_tfr_type) const { in CastToNonDerivedType()251 if (auto cast_op = dyn_cast_or_null<CastOp>(input_op)) { in CollectInputsAndAttributes() local273 auto cast_op = dyn_cast_or_null<CastOp>(list_input.getDefiningOp()); in CollectInputsAndAttributes() local
111 auto* cast_op = static_cast<CastOperator*>(op); in Run() local
188 CastOperator const* cast_op = static_cast<CastOperator const*>(unary_op); in Run() local
129 LogicalResult HandleCast(TF::CastOp cast_op, ArrayRef<int64_t> new_shape) { in HandleCast()184 if (auto cast_op = dyn_cast_or_null<TF::CastOp>(input.getDefiningOp())) in HandlePad() local550 auto cast_op = dyn_cast_or_null<TF::CastOp>(next_input.getDefiningOp()); in GetInputBlockArgNum() local
223 while (auto cast_op = in MatchCallArgs() local
645 TF::CastOp cast_op; in UpdateTypeAndInsertIncompatibleUseCasts() local1060 if (auto cast_op = dyn_cast<CastOp>(op)) return InferShapeForCast(cast_op); in InferShapeForSingleOperation() local1481 if (auto cast_op = dyn_cast_or_null<CastOp>(arg_defining_op)) { in InferShapeForFunctionReturnType() local
167 auto cast_op = in ReplaceAllUsesWithCast() local
442 if (auto cast_op = llvm::dyn_cast_or_null<CastOp>(preceding_op)) { in matchAndRewrite() local
216 const string cast_op = (host_memory) ? "_HostCast" : "Cast"; in AddSend() local313 const string cast_op = (host_memory) ? "_HostCast" : "Cast"; in AddRecv() local
868 Value cast_op = nested_builder.create<IndexCastOp>( in matchAndRewrite() local
1189 tensorflow::NodeDef* cast_op = tensorflow_graph->add_node(); in ConvertCastOperator() local