Home
last modified time | relevance | path

Searched refs:IteratorGetNextOp (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h200 data::IteratorGetNextOp); \
204 data::IteratorGetNextOp); \
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.h234 class IteratorGetNextOp : public HybridAsyncOpKernel {
236 explicit IteratorGetNextOp(OpKernelConstruction* ctx) in IteratorGetNextOp() function
Diterator_ops.cc952 AsyncOpKernel* IteratorGetNextOp::AsAsync() { in AsAsync()
967 Status IteratorGetNextOp::DoCompute(OpKernelContext* ctx) { in DoCompute()
1205 IteratorGetNextOp);
1207 IteratorGetNextOp);
1210 IteratorGetNextOp);
1213 IteratorGetNextOp);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc95 TF::IteratorGetNextOp iterator_op = llvm::dyn_cast<TF::IteratorGetNextOp>(op); in IsSupportedInputOp()
Dtpu_space_to_depth_pass.cc410 TF::IteratorGetNextOp iter = llvm::dyn_cast<TF::IteratorGetNextOp>(op); in IsSupportedHostInputOp()