Searched refs:operation_name (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 88 void StartExecute(TFE_Context* context, const char* operation_name,
99 void Execute(TFE_Context* context, const char* operation_name,
176 const char* operation_name, in StartExecute() argument
189 operation_name_ = operation_name; in StartExecute()
221 void DeviceThread::Execute(TFE_Context* context, const char* operation_name, in Execute() argument
229 op_.reset(TFE_NewOp(context, operation_name, status)); in Execute()
234 TFE_OpReset(op_.get(), operation_name, device_.c_str(), status); in Execute()
299 const char* operation_name, in Execute() argument
303 StartExecute(context, inputs, operation_name, attributes, in Execute()
322 const char* operation_name, const TFE_OpAttrs* attributes, in StartExecute() argument
[all …]
|
| D | parallel_device.cc | 67 std::vector<MaybeParallelTensorUnowned> inputs, const char* operation_name, in ExecuteWithSpecialOps() argument
73 if (operation_name == std::string("TPUReplicatedInput")) { in ExecuteWithSpecialOps()
107 } else if (operation_name == std::string("TPUReplicatedOutput")) { in ExecuteWithSpecialOps()
110 OpPtr op(TFE_NewOp(context, operation_name, status)); in ExecuteWithSpecialOps()
166 parallel_device.Execute(context, parallel_inputs, operation_name, in ExecuteWithSpecialOps()
290 const char* operation_name = TFE_OpGetName(original_op, status); in ParallelDeviceExecute() local
319 context, std::move(typed_inputs), operation_name, in ParallelDeviceExecute()
|
| D | parallel_device_lib.h | 97 const char* operation_name, const TFE_OpAttrs* attributes,
114 const char* operation_name, const TFE_OpAttrs* attributes,
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | custom_device_testutil.cc | 105 const char* operation_name = TFE_OpGetName(original_op, s); in LoggingDeviceExecute() local
109 TFE_Op* op(TFE_NewOp(context, operation_name, s)); in LoggingDeviceExecute()
|
| /external/antlr/runtime/Ruby/lib/antlr3/streams/ |
| D | rewrite.rb | 84 self.class.operation_name
|