Searched refs:ExitOp (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | control_flow_ops.h | 93 class ExitOp : public OpKernel {
95 explicit ExitOp(OpKernelConstruction* context) : OpKernel(context) {} in ExitOp() function
98 ~ExitOp() override {} in ~ExitOp()
100 TF_DISALLOW_COPY_AND_ASSIGN(ExitOp);
|
| D | control_flow_ops.cc | 352 void ExitOp::Compute(OpKernelContext* context) { in Compute()
360 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_DEFAULT), ExitOp);
361 REGISTER_KERNEL_BUILDER(Name("Exit").Device(DEVICE_TPU_SYSTEM), ExitOp);
362 REGISTER_KERNEL_BUILDER(Name("RefExit").Device(DEVICE_CPU), ExitOp);
366 Name("Exit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp);
369 Name("RefExit").Device(DEVICE_GPU).TypeConstraint<type>("T"), ExitOp);
390 ExitOp); \
396 ExitOp)
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_device_propagation.cc | 78 return llvm::isa<tf_executor::EnterOp, tf_executor::ExitOp, in IsSupportedExecutorOp()
|
| D | shape_inference.cc | 115 tf_executor::EnterOp, tf_executor::ExitOp, tf_executor::FetchOp, in IsSupportedNonTFOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | tf_to_tfl_flatbuffer.cc | 54 mlir::tf_executor::EnterOp, mlir::tf_executor::ExitOp, in IsControlFlowV1Op()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 842 void Print(ExitOp exit, OpAsmPrinter &p) { in Print()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1773 return builder_.create<mlir::tf_executor::ExitOp>(loc, types, operands, in CreateOperation()
|