Searched refs:cast_builder (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/cc/ops/ |
| D | const_op.h | 57 auto cast_builder = NodeBuilder(cast_op_name, "Cast") in Const() local
60 scope.UpdateBuilder(&cast_builder); in Const()
62 scope.UpdateStatus(cast_builder.Finalize(scope.graph(), &ret)); in Const()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | op_builder.cc | 253 OpBuilder* cast_builder = CreateCastBuilder(this, op_type); in AddCastOp() local
254 builders_.emplace_back(cast_builder); in AddCastOp()
255 cast_builder->SetNodeId(builders_.size()); in AddCastOp()
262 cast_builder->PopulateSubGraph(tensor_data, tensor_data, context)); in AddCastOp()
263 TF_LITE_ENSURE_STATUS(cast_builder->RegisterOutputs(tensor_data, context)); in AddCastOp()
266 if (cast_op_builder != nullptr) *cast_op_builder = cast_builder; in AddCastOp()
|
| D | BUILD | 15 "cast_builder.cc",
49 "cast_builder.h",
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 217 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op, in AddSend() local
219 cast_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
221 cast_builder.Attr("_start_time", start_time); in AddSend()
223 cast_builder.Attr("DstT", cast_dtype); in AddSend()
229 cast_builder.Attr("Truncate", true); in AddSend()
233 *status = cast_builder.Finalize(cast, /*consume=*/true); in AddSend()
314 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op, in AddRecv() local
316 cast_builder.Attr("DstT", dtype); in AddRecv()
317 cast_builder.Device(dst->assigned_device_name()) in AddRecv()
320 *status = cast_builder.Finalize(cast, /*consume=*/true); in AddRecv()
|