Searched refs:pack_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | convert_trivial_pack_to_reshape.cc | 36 auto* pack_op = static_cast<PackOperator*>(pack_it->get()); in Run() local
37 if (pack_op->inputs.size() > 1) { in Run()
41 CHECK_EQ(pack_op->outputs.size(), 1); in Run()
43 const auto& input_array = model->GetArray(pack_op->inputs[0]); in Run()
54 AddMessageF("Converting trivial %s to a reshape", LogName(*pack_op)); in Run()
58 reshape_op->inputs = {pack_op->inputs[0]}; in Run()
59 reshape_op->outputs = pack_op->outputs; in Run()
63 AvailableArrayName(*model, pack_op->outputs[0] + "_shape"); in Run()
78 CHECK_EQ(pack_it->get(), pack_op); in Run()
|
| D | unroll_batch_matmul.cc | 147 auto* pack_op = new PackOperator; in UnrollBatchMatMulRecursion() local
150 pack_op->inputs = pack_inputs; in UnrollBatchMatMulRecursion()
151 pack_op->outputs = {AvailableArrayName(*model, batch_name + "/pack")}; in UnrollBatchMatMulRecursion()
152 auto& pack_op_output = model->GetOrCreateArray(pack_op->outputs[0]); in UnrollBatchMatMulRecursion()
154 pack_op->axis = 0; in UnrollBatchMatMulRecursion()
155 pack_op->values_count = pack_inputs.size(); in UnrollBatchMatMulRecursion()
156 *tail_it = model->operators.emplace(*tail_it, pack_op) + 1; in UnrollBatchMatMulRecursion()
158 batch_pack_inputs.push_back(pack_op->outputs[0]); in UnrollBatchMatMulRecursion()
285 auto* pack_op = new PackOperator; in Run() local
286 pack_op->inputs = pack_inputs; in Run()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1335 tensorflow::NodeDef* pack_op = tensorflow_graph->add_node(); in ConvertPackOperator() local
1336 pack_op->set_op("Pack"); in ConvertPackOperator()
1337 pack_op->set_name(src_op.outputs[0]); in ConvertPackOperator()
1339 *pack_op->add_input() = input; in ConvertPackOperator()
1341 (*pack_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertPackOperator()
1342 (*pack_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertPackOperator()
1343 (*pack_op->mutable_attr())["T"].set_type( in ConvertPackOperator()
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 253 pack_op = array_ops.stack(
257 return core.LabeledTensor(pack_op, axes)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 61 "pack_op.cc",
|
| /external/tensorflow/tensorflow/contrib/makefile/ |
| D | tf_op_files.txt | 168 tensorflow/core/kernels/pack_op.cc
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 894 ":pack_op",
1033 name = "pack_op",
1034 prefix = "pack_op",
5523 "pack_op.cc",
|