Home
last modified time | relevance | path

Searched refs:EnterOp (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcontrol_flow_ops.h80 class EnterOp : public OpKernel {
82 explicit EnterOp(OpKernelConstruction* context) : OpKernel(context) {} in EnterOp() function
85 ~EnterOp() override {} in ~EnterOp()
87 TF_DISALLOW_COPY_AND_ASSIGN(EnterOp);
Dcontrol_flow_ops.cc295 void EnterOp::Compute(OpKernelContext* context) { in Compute()
303 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_DEFAULT), EnterOp);
304 REGISTER_KERNEL_BUILDER(Name("Enter").Device(DEVICE_TPU_SYSTEM), EnterOp);
305 REGISTER_KERNEL_BUILDER(Name("RefEnter").Device(DEVICE_CPU), EnterOp);
309 Name("Enter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)
312 Name("RefEnter").Device(DEVICE_GPU).TypeConstraint<type>("T"), EnterOp)
333 EnterOp)
341 EnterOp)
/external/tensorflow/tensorflow/core/profiler/utils/
Dop_utils.h41 void EnterOp(absl::string_view name, absl::string_view category,
70 void EnterOp(uint64 program_id, absl::string_view name,
Dop_utils.cc43 void HostOpMetricsDbBuilder::EnterOp(absl::string_view name, in EnterOp() function in tensorflow::profiler::HostOpMetricsDbBuilder
67 void DeviceOpMetricsDbBuilder::EnterOp( in EnterOp() function in tensorflow::profiler::DeviceOpMetricsDbBuilder
/external/tensorflow/tensorflow/core/profiler/convert/
Dxplane_to_op_metrics_db.cc97 tf_metrics_data->tf_metrics_db_builder.EnterOp( in ProcessOneTfActivity()
244 device_op_metrics_db_builder.EnterOp( in ConvertDeviceTraceXPlaneToOpMetricsDb()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_device_propagation.cc78 return llvm::isa<tf_executor::EnterOp, tf_executor::ExitOp, in IsSupportedExecutorOp()
Dshape_inference.cc115 tf_executor::EnterOp, tf_executor::ExitOp, tf_executor::FetchOp, in IsSupportedNonTFOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc54 mlir::tf_executor::EnterOp, mlir::tf_executor::ExitOp, in IsControlFlowV1Op()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor.cc706 void Print(EnterOp enter, OpAsmPrinter &p) { in Print()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1769 return builder_.create<mlir::tf_executor::EnterOp>(loc, types, operands, in CreateOperation()