Searched refs:CoreRuntimeOp (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | op_cache.cc | 25 Expected<CoreRuntimeOp*> OpCache::GetOrAddOp( in GetOrAddOp()
47 Expected<CoreRuntimeOp> expected_op = in GetOrAddOp()
59 Expected<CoreRuntimeOp*> OpCache::GetOrAddXlaOp(string_view op_name, in GetOrAddXlaOp()
70 Expected<CoreRuntimeOp> expected_op = context->GetCoreRuntime()->MakeOp( in GetOrAddXlaOp()
|
| D | function_cache.h | 48 BefBuffer bef_buffer, RCReference<BEFFile> bef_file, CoreRuntimeOp fn, in CreateFunctionState()
56 const CoreRuntimeOp& GetFunc() const { return fn_; } in GetFunc()
69 BefBuffer bef_buffer, RCReference<BEFFile> bef_file, CoreRuntimeOp fn, in FunctionState()
82 const CoreRuntimeOp fn_;
|
| D | op_cache.h | 47 Expected<CoreRuntimeOp*> GetOrAddOp(string_view op_name,
57 Expected<CoreRuntimeOp*> GetOrAddXlaOp(string_view op_name,
182 std::unordered_map<CacheKey, CoreRuntimeOp, CacheKeyHash> cache_
|
| D | c_api_tfrt.h | 58 class CoreRuntimeOp; variable
595 CoreRuntimeOp* op_;
|
| D | c_api_tfrt.cc | 1444 Expected<CoreRuntimeOp*> expected_op = context_->GetOpCache().GetOrAddOp( in Initialize()
1465 Expected<CoreRuntimeOp*> expected_op = in Initialize()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_op_handler.cc | 62 llvm::Expected<tfrt::CoreRuntimeOp> MakeOp(
92 using tfrt::CoreRuntimeOp;
223 Expected<CoreRuntimeOp> RuntimeFallbackOpHandler::MakeOp(string_view op_name) { in MakeOp()
233 return CoreRuntimeOp( in MakeOp()
|
| D | runtime_fallback_batch_kernels.cc | 41 using ::tfrt::CoreRuntimeOp;
49 void ExecuteFallbackOp(CoreRuntimeOp op, in ExecuteFallbackOp()
|
| D | kernel_utils.cc | 91 tfrt::Expected<tfrt::CoreRuntimeOp> GetFallbackOp(tfrt::string_view op_name, in GetFallbackOp()
|
| D | kernel_utils.h | 105 llvm::Expected<tfrt::CoreRuntimeOp> GetFallbackOp(tfrt::string_view op_name,
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_op_handler.cc | 45 llvm::Expected<tfrt::CoreRuntimeOp> MakeOp(
71 using tfrt::CoreRuntimeOp;
233 Expected<CoreRuntimeOp> KernelFallbackOpHandler::MakeOp(string_view op_name) { in MakeOp()
240 return CoreRuntimeOp( in MakeOp()
|