Searched refs:begin_op (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 134 mlir::Operation* HoistAndFix(llvm::iplist<mlir::Operation>::iterator begin_op, in HoistAndFix() argument
143 if (&ancestors[i].getBody()->front() == &*begin_op) { in HoistAndFix()
158 if (auto alloc = mlir::dyn_cast<mlir::AllocOp>(begin_op)) { in HoistAndFix()
159 CHECK(std::next(begin_op) == end_op) in HoistAndFix()
196 for (mlir::Operation& op : llvm::make_range(begin_op, end_op)) { in HoistAndFix()
214 llvm::make_early_inc_range(llvm::make_range(begin_op, end_op))) { in HoistAndFix()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1635 auto begin_op = begin.getDefiningOp(); in matchAndRewrite() local
1638 if (begin_op == nullptr && size_op == nullptr) return failure(); in matchAndRewrite()
1643 if (begin_op && begin_type && begin_type.getElementType().isInteger(64)) { in matchAndRewrite()
1645 begin_op, begin_type, slice_op.getLoc(), &rewriter); in matchAndRewrite()
|
| /external/grpc-grpc/src/core/lib/surface/ |
| D | completion_queue.cc | 191 bool (*begin_op)(grpc_completion_queue* cq, void* tag); member
673 return cq->vtable->begin_op(cq, tag);
|
| /external/rust/crates/grpcio-sys/grpc/src/core/lib/surface/ |
| D | completion_queue.cc | 204 bool (*begin_op)(grpc_completion_queue* cq, void* tag); member
686 return cq->vtable->begin_op(cq, tag);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 72 auto begin_op = ops::Const(root, {begin}); in TestStridedSlice() local
75 auto slice = ops::StridedSlice(root, input, begin_op, end_op, stride_op, in TestStridedSlice()
87 TF_ASSERT_OK(m.AddNode(begin_op.node())); in TestStridedSlice()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 706 auto begin_op = rewriter.create<ConstantOp>(op->getLoc(), type, begin_attr); in RewriteEllipsisMask() local
714 op, strided_slice_op.getType(), input, begin_op.getResult(), in RewriteEllipsisMask()
|