Home
last modified time | relevance | path

Searched defs:next_op (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_squeeze_attributes.cc41 const auto* next_op = GetOpWithInput(*model, squeeze_op->outputs[0]); in Run() local
Dremove_trivial_reshape.cc60 const auto* next_op = GetOpWithInput(model, op.outputs[0]); in IsReshapeTrivial() local
Didentify_dilated_conv.cc97 auto* next_op = has_pad_op in ResolveDilatedConv() local
Didentify_nearest_upsample.cc96 Operator* next_op = GetOpWithOutput(*model, output); in Run() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddilated_conv.h109 Operation* next_op = op.getOperation()->getNextNode(); in matchAndRewrite() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_ops_test.cc232 Output next_op = ops::Conv2D( in RunConv2DUnfused() local
562 Output next_op = ops::DepthwiseConv2dNative( in RunDepthwiseConv2DUnfused() local
1068 Output next_op = ops::MatMul(root.WithOpName("matmul"), input_op, in VerifyFusedMatMul() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc769 Operation* next_op = curr_op->getNextNode(); in RewriteFunction() local