Searched refs:pack_op (Results 1 – 10 of 10) 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()
74 dim == pack_op->axis in Run()
81 DeleteOpAndArrays(model, pack_op); in Run()
|
| D | unroll_batch_matmul.cc | 234 auto* pack_op = new PackOperator; in Run() local
235 pack_op->inputs = pack_inputs; in Run()
236 pack_op->outputs = {AvailableArrayName(*model, base_name + "/pack")}; in Run()
237 auto& pack_op_output = model->GetOrCreateArray(pack_op->outputs[0]); in Run()
239 pack_op->axis = 0; in Run()
240 pack_op->values_count = pack_inputs.size(); in Run()
241 tail_it = model->operators.emplace(tail_it, pack_op) + 1; in Run()
255 pack_op->outputs[0], in Run()
|
| D | identify_nearest_upsample.cc | 229 auto* pack_op = new PackOperator; in Run() local
232 pack_op->inputs.push_back(current_pack_input_name); in Run()
236 pack_op->axis = axis; in Run()
237 pack_op->values_count = scale; in Run()
257 pack_op->outputs.push_back(pack_output_array_name); in Run()
261 to_be_inserted_ops.push_back(pack_op); in Run()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 292 auto pack_op = in matchAndRewrite() local
303 auto reshape_op = createReshapeOp(pack_op.output(), result_shape, in matchAndRewrite()
|
| D | shape_inference.cc | 346 if (auto pack_op = dyn_cast<TF::PackOp>(op)) { in ComputeInputsRequiredForOutput() local
347 auto type = pack_op.getType().cast<TensorType>(); in ComputeInputsRequiredForOutput()
351 ValuePort req(pack_op.getOperand(port[1])); in ComputeInputsRequiredForOutput()
383 if (auto pack_op = dyn_cast<TF::PackOp>(op)) { in ComputeOutputComponent() local
384 TensorType type = pack_op.getType().cast<TensorType>(); in ComputeOutputComponent()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute.cc | 764 EagerOperation pack_op(&ctx); in MaybePackInputTensor() local
765 TF_RETURN_IF_ERROR(pack_op.Reset("Pack", /*device_name=*/nullptr, in MaybePackInputTensor()
767 pack_op.MutableAttrs()->Set("N", handle->NumPackedHandles()); in MaybePackInputTensor()
768 pack_op.MutableAttrs()->Set("T", handle->dtype); in MaybePackInputTensor()
772 TF_RETURN_IF_ERROR(pack_op.AddInput(h)); in MaybePackInputTensor()
777 EagerLocalExecute(&pack_op, retvals.data(), &num_retvals)); in MaybePackInputTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1497 TFL::PackOp pack_op = cast<TFL::PackOp>(op); in matchAndRewrite() local
1498 Operation *first_input = pack_op.getOperand(0).getDefiningOp(); in matchAndRewrite()
1504 if (pack_op.axis() != input_unpack_op.axis() || in matchAndRewrite()
1505 pack_op.values_count() != input_unpack_op.num()) in matchAndRewrite()
1508 const int total_pack_inputs = pack_op.getNumOperands(); in matchAndRewrite()
1512 llvm::zip(pack_op.getOperands(), input_unpack_op.getResults())) { in matchAndRewrite()
1520 rewriter.replaceOp(pack_op, input_unpack_op.getOperand()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1348 tensorflow::NodeDef* pack_op = tensorflow_graph->add_node(); in ConvertPackOperator() local
1349 pack_op->set_op("Pack"); in ConvertPackOperator()
1350 pack_op->set_name(src_op.outputs[0]); in ConvertPackOperator()
1352 *pack_op->add_input() = input; in ConvertPackOperator()
1354 (*pack_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertPackOperator()
1355 (*pack_op->mutable_attr())["N"].set_i(src_op.inputs.size()); in ConvertPackOperator()
1356 (*pack_op->mutable_attr())["T"].set_type( in ConvertPackOperator()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | BUILD | 73 "pack_op.cc",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 976 ":pack_op",
1113 name = "pack_op",
1114 prefix = "pack_op",
5827 "pack_op.cc",
|