Home
last modified time | relevance | path

Searched refs:next_op (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddilated_conv.h109 Operation* next_op = op.getOperation()->getNextNode(); in matchAndRewrite() local
110 if (!next_op) return failure(); in matchAndRewrite()
117 if (!llvm::isa<TF::SqueezeOp>(next_op)) { in matchAndRewrite()
122 squeeze_op = llvm::cast<TF::SqueezeOp>(next_op); in matchAndRewrite()
156 next_op = next_op->getNextNode(); in matchAndRewrite()
157 if (!next_op) return failure(); in matchAndRewrite()
170 if (llvm::isa<TF::PadOp>(next_op)) { in matchAndRewrite()
171 pad_op = llvm::cast<TF::PadOp>(next_op); in matchAndRewrite()
172 next_op = next_op->getNextNode(); in matchAndRewrite()
173 if (!next_op) return failure(); in matchAndRewrite()
[all …]
/external/tensorflow/tensorflow/python/distribute/
Dtest_util.py140 for next_op in _op_dependencies(op):
141 in_degrees[next_op] = in_degrees.get(next_op, 0) + 1
152 for next_op in _op_dependencies(op):
153 in_degrees[next_op] -= 1
154 if in_degrees[next_op] == 0:
155 nexts.append(next_op)
165 for next_op in _op_dependencies(op):
166 if next_op == end:
168 nexts.append(next_op)
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dremove_trivial_reshape.cc60 const auto* next_op = GetOpWithInput(model, op.outputs[0]); in IsReshapeTrivial() local
61 if (next_op->type == OperatorType::kReshape) { in IsReshapeTrivial()
62 if (!IsDiscardableArray(model, next_op->outputs[0])) { in IsReshapeTrivial()
68 LogName(op), LogName(*next_op)); in IsReshapeTrivial()
74 LogName(op), LogName(*next_op)); in IsReshapeTrivial()
Didentify_dilated_conv.cc97 auto* next_op = has_pad_op in ResolveDilatedConv() local
103 if (next_op->type == OperatorType::kAdd) { in ResolveDilatedConv()
106 auto final_op = GetOpWithInput(*model, next_op->outputs[0]); in ResolveDilatedConv()
109 const auto* bts_op = has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv()
117 Operator* bias_add_op = !has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv()
Dresolve_squeeze_attributes.cc41 const auto* next_op = GetOpWithInput(*model, squeeze_op->outputs[0]); in Run() local
42 if (next_op->type == OperatorType::kReshape) { in Run()
Didentify_nearest_upsample.cc96 Operator* next_op = GetOpWithOutput(*model, output); in Run() local
97 if (next_op == nullptr) { in Run()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_ops_test.cc232 Output next_op = ops::Conv2D( in RunConv2DUnfused() local
242 next_op = ops::BiasAdd( in RunConv2DUnfused()
243 root.WithOpName(last_op), next_op, in RunConv2DUnfused()
250 next_op = ops::AddN(root.WithOpName("with_add"), in RunConv2DUnfused()
251 std::initializer_list<Input>{next_op, input_data_op}); in RunConv2DUnfused()
257 next_op = ops::Relu(root.WithOpName(last_op), next_op); in RunConv2DUnfused()
263 next_op = ops::Relu6(root.WithOpName(last_op), next_op); in RunConv2DUnfused()
269 next_op = ops::Elu(root.WithOpName(last_op), next_op); in RunConv2DUnfused()
275 next_op = ops::internal::LeakyRelu(root.WithOpName(last_op), next_op); in RunConv2DUnfused()
562 Output next_op = ops::DepthwiseConv2dNative( in RunDepthwiseConv2DUnfused() local
[all …]
/external/llvm-project/llvm/test/Transforms/CodeGenPrepare/X86/
Dgep-unmerging.ll52 %next_op = load i8, i8* %p_preinc
53 %p_zext = zext i8 %next_op to i64
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc769 Operation* next_op = curr_op->getNextNode(); in RewriteFunction() local
814 ops_to_visit.push_back({/*region_idx=*/llvm::None, token, next_op}); in RewriteFunction()