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

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
convert_trivial_pack_to_reshape.cc 36 auto* pack_op = static_cast<PackOperator*>(pack_it->get()); local
37 if (pack_op->inputs.size() > 1) {
41 CHECK_EQ(pack_op->outputs.size(), 1);
43 const auto& input_array = model->GetArray(pack_op->inputs[0]);
54 AddMessageF("Converting trivial %s to a reshape", LogName(*pack_op));
58 reshape_op->inputs = {pack_op->inputs[0]};
59 reshape_op->outputs = pack_op->outputs;
63 AvailableArrayName(*model, pack_op->outputs[0] + "_shape");
78 CHECK_EQ(pack_it->get(), pack_op);
unroll_batch_matmul.cc 147 auto* pack_op = new PackOperator; local
150 pack_op->inputs = pack_inputs;
151 pack_op->outputs = {AvailableArrayName(*model, batch_name + "/pack")};
152 auto& pack_op_output = model->GetOrCreateArray(pack_op->outputs[0]);
154 pack_op->axis = 0;
155 pack_op->values_count = pack_inputs.size();
156 *tail_it = model->operators.emplace(*tail_it, pack_op) + 1;
158 batch_pack_inputs.push_back(pack_op->outputs[0]);
285 auto* pack_op = new PackOperator; local
286 pack_op->inputs = pack_inputs
    [all...]
  /external/tensorflow/tensorflow/lite/toco/
export_tensorflow.cc 1335 tensorflow::NodeDef* pack_op = tensorflow_graph->add_node(); local
    [all...]
  /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
ops.py 253 pack_op = array_ops.stack(
257 return core.LabeledTensor(pack_op, axes)
    [all...]

Completed in 96 milliseconds