Home
last modified time | relevance | path

Searched refs:cast_op (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_unary.cc187 CastOperator const* cast_op = static_cast<CastOperator const*>(unary_op); in Run() local
188 if (cast_op->dst_data_type != ArrayDataType::kFloat) { in Run()
195 if (cast_op->src_data_type != input_array.buffer->type) { in Run()
Dpropagate_array_data_types.cc106 auto* cast_op = static_cast<CastOperator*>(op); in Run() local
107 model->GetArray(op->outputs[0]).data_type = cast_op->dst_data_type; in Run()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc212 const string cast_op = (host_memory) ? "_HostCast" : "Cast"; in AddSend() local
213 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op, in AddSend()
286 const string cast_op = (host_memory) ? "_HostCast" : "Cast"; in AddRecv() local
287 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op, in AddRecv()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
DBUILD159 "//tensorflow/core/kernels:cast_op",
705 "//tensorflow/core/kernels:cast_op",
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1186 tensorflow::NodeDef* cast_op = tensorflow_graph->add_node(); in ConvertCastOperator() local
1187 cast_op->set_op("Cast"); in ConvertCastOperator()
1188 cast_op->set_name(src_op.outputs[0]); in ConvertCastOperator()
1190 *cast_op->add_input() = src_op.inputs[0]; in ConvertCastOperator()
1192 (*cast_op->mutable_attr())["DstT"].set_type( in ConvertCastOperator()
1194 (*cast_op->mutable_attr())["SrcT"].set_type( in ConvertCastOperator()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD22 "cast_op.cc",
/external/tensorflow/tensorflow/
Dtensorflow.bzl1282 For example, with prefix = "cast_op",
1283 * srcs = ["cast_op.cc"]
1284 * hdrs = ["cast_op.h"]
1285 * gpu_srcs = ["cast_op_gpu.cu.cc", "cast_op.h"]
/external/tensorflow/tensorflow/core/kernels/
DBUILD3026 ":cast_op",
3262 ":cast_op",
3318 name = "cast_op",
3319 prefix = "cast_op",
3455 ":cast_op",
5459 "cast_op.cc",
5460 "cast_op.h",
/external/tensorflow/tensorflow/core/distributed_runtime/
DBUILD119 "//tensorflow/core/kernels:cast_op",
/external/tensorflow/tensorflow/core/
DBUILD1628 "//tensorflow/core/kernels:cast_op",
4341 "//tensorflow/core/kernels:cast_op",
4400 "//tensorflow/core/kernels:cast_op",
4456 "//tensorflow/core/kernels:cast_op",
4715 "//tensorflow/core/kernels:cast_op",
4751 "//tensorflow/core/kernels:cast_op",
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_op_files.txt21 tensorflow/core/kernels/cast_op.cc
/external/tensorflow/tensorflow/compiler/jit/
DBUILD196 "//tensorflow/core/kernels:cast_op",