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

  /external/tensorflow/tensorflow/core/kernels/hexagon/
graph_transferer.cc 150 graph_input_node_info.add_shape(dim);
175 graph_output_node_info.add_shape(dim);
482 const_node_info.add_shape(shape_array[0]);
483 const_node_info.add_shape(shape_array[1]);
484 const_node_info.add_shape(shape_array[2]);
485 const_node_info.add_shape(shape_array[3]);
513 const_node_info.add_shape(static_cast<int64>(shape[0]));
514 const_node_info.add_shape(static_cast<int64>(shape[1]));
515 const_node_info.add_shape(static_cast<int64>(shape[2]));
516 const_node_info.add_shape(static_cast<int64>(shape[3]))
    [all...]
  /external/tensorflow/tensorflow/core/framework/
attr_value_util.cc 353 v.AsProto(out->mutable_list()->add_shape());
360 *out->mutable_list()->add_shape() = v;
368 v.AsProto(out->mutable_list()->add_shape());
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_tfgraph_builder.cc 157 *shapes.mutable_list()->add_shape() = GetTensorShape(instruction);
  /external/tensorflow/tensorflow/tools/graph_transforms/
fold_constants_test.cc 145 auto* shape_proto = shape_attr.mutable_list()->add_shape();
  /external/tensorflow/tensorflow/core/profiler/internal/
tfprof_node.h 387 node_.add_shape(s);
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties.cc     [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
layout_optimizer.cc 807 auto output_shape = attr_output_shape.mutable_list()->add_shape();
    [all...]
  /external/tensorflow/tensorflow/contrib/lite/schema/
schema_generated.h 1274 void add_shape(flatbuffers::Offset<flatbuffers::Vector<int32_t>> shape) { function in struct:tflite::TensorBuilder
    [all...]

Completed in 1004 milliseconds