Home
last modified time | relevance | path

Searched defs:pack_op (Results 1 – 10 of 10) 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/mesa3d/src/amd/common/nir/
Dac_nir_lower_ps_late.c419 nir_op pack_op = nir_op_pack_32_2x16; in emit_ps_color_export() 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