Searched refs:cast_builder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op.h | 55 auto cast_builder = NodeBuilder(cast_op_name, "Cast") in Const() local
58 scope.UpdateBuilder(&cast_builder); in Const()
60 scope.UpdateStatus(cast_builder.Finalize(scope.graph(), &ret)); in Const()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 213 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op, in AddSend() local
215 cast_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
217 cast_builder.Attr("_start_time", start_time); in AddSend()
219 cast_builder.Attr("DstT", cast_dtype); in AddSend()
225 cast_builder.Attr("Truncate", true); in AddSend()
229 *status = cast_builder.Finalize(cast); in AddSend()
287 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op, in AddRecv() local
289 cast_builder.Attr("DstT", dtype); in AddRecv()
290 cast_builder.Device(dst->assigned_device_name()) in AddRecv()
293 *status = cast_builder.Finalize(cast); in AddRecv()
|