Searched refs:op_filter (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass.cc | 168 const OperationFilter& op_filter, int depth,
176 const OperationFilter& op_filter, int depth, in IsCompilableWhile() argument
191 if (!IsCompilableCall(call, jit_device_type, op_filter, depth + 1, in IsCompilableWhile()
207 if (!IsCompilableCall(call, jit_device_type, op_filter, depth + 1, in IsCompilableWhile()
221 const OperationFilter& op_filter, int depth, in IsCompilableCall() argument
261 return IsCompilableWhile(*node, jit_device_type, op_filter, depth + 1, in IsCompilableCall()
264 if (!op_filter.allow_resource_ops && in IsCompilableCall()
268 if (!op_filter.allow_stateful_rng_ops && in IsCompilableCall()
272 if (!op_filter.allow_control_trigger && node->IsControlTrigger()) { in IsCompilableCall()
275 if (!op_filter.allow_dummy_ops && IsDummyImplOp(node->type_string())) { in IsCompilableCall()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_registry.cc | 120 absl::Span<const DataType> supported_types, BackendOpFilter op_filter) { in RegisterBackend() argument
128 result.first->second.op_filter = op_filter; in RegisterBackend()
309 if (backend.second.op_filter != nullptr && in RegisterCompilationKernels()
310 !backend.second.op_filter(kdef.get())) { in RegisterCompilationKernels()
554 XlaOpRegistry::BackendOpFilter op_filter) { in XlaBackendRegistrar() argument
556 registry.RegisterBackend(string(name), types, op_filter); in XlaBackendRegistrar()
|
| D | xla_op_registry.h | 107 BackendOpFilter op_filter);
189 BackendOpFilter op_filter; member
344 XlaOpRegistry::BackendOpFilter op_filter = nullptr);
|