HomeSort by relevance Sort by last modified time
    Searched refs:cast_op (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
propagate_array_data_types.cc 77 auto* cast_op = static_cast<CastOperator*>(op); local
78 model->GetArray(op->outputs[0]).data_type = cast_op->dst_data_type;
resolve_constant_unary.cc 85 CastOperator const* cast_op = static_cast<CastOperator const*>(unary_op); local
86 if (cast_op->dst_data_type != ArrayDataType::kFloat) {
93 if (cast_op->src_data_type != input_array.buffer->type) {
  /external/tensorflow/tensorflow/core/graph/
graph_partition.cc 209 const string cast_op = (host_memory) ? "_HostCast" : "Cast"; local
210 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op);
272 const string cast_op = (host_memory) ? "_HostCast" : "Cast"; local
273 NodeDefBuilder cast_builder(opts.new_name(src->name()), cast_op);
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/toco/
export_tensorflow.cc 1014 auto* cast_op = tensorflow_graph->add_node(); local
    [all...]

Completed in 120 milliseconds