Home
last modified time | relevance | path

Searched defs:pack_op (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_trivial_pack_to_reshape.cc36 auto* pack_op = static_cast<PackOperator*>(pack_it->get()); in Run() local
Didentify_nearest_upsample.cc230 auto* pack_op = new PackOperator; in Run() local
Dunroll_batch_matmul.cc237 auto* pack_op = new PackOperator; in Run() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc270 auto pack_op = in matchAndRewrite() local
Dshape_inference.cc495 if (auto pack_op = dyn_cast<TF::PackOp>(op)) { in ComputeInputsRequiredForOutput() local
538 if (auto pack_op = dyn_cast<TF::PackOp>(op)) { in ComputeOutputComponent() local
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc207 TFL::PackOp pack_op, PatternRewriter& rewriter) const { in matchAndRewrite()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc1473 EagerOperation pack_op(&ctx); in MaybePackInputTensor() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1937 TFL::PackOp pack_op = cast<TFL::PackOp>(op); in matchAndRewrite() local
1974 TFL::PackOp pack_op = cast<TFL::PackOp>(op); in matchAndRewrite() local
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1353 tensorflow::NodeDef* pack_op = tensorflow_graph->add_node(); in ConvertPackOperator() local