| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device.cc | 155 bool ShouldLogInputsAndOutputs(OpKernel* op_kernel) { in ShouldLogInputsAndOutputs()
162 void ThreadPoolDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
176 void ThreadPoolDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
193 void ThreadPoolDevice::LogInputs(OpKernel* op_kernel, in LogInputs()
208 void ThreadPoolDevice::LogOutputs(OpKernel* op_kernel, in LogOutputs()
|
| D | renamed_device.h | 122 void Compute(OpKernel* op_kernel, OpKernelContext* context) override { in Compute()
126 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| D | graph_view.cc | 430 const OpKernel* op_kernel = item->kernel; in SetAllocAttrs() local
|
| D | single_threaded_executor.cc | 220 OpKernel* op_kernel = kernel_state.kernel; in Initialize() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device.h | 87 virtual void Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
92 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| D | op_kernel_test.cc | 826 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
847 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
900 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
929 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
941 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
982 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
992 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
|
| D | op_kernel.h | 562 OpKernel* op_kernel = nullptr; member
686 const OpKernel& op_kernel() const { return *params_->op_kernel; } in op_kernel() function
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 91 OpKernel* op_kernel, XlaContext& context) { in AttachLocationToMetadata()
101 void XlaCompilationDevice::Compute(OpKernel* op_kernel, in Compute()
|
| D | const_analysis.cc | 89 Status GetCompileTimeConstInputs(const NodeDef& node, const OpKernel* op_kernel, in GetCompileTimeConstInputs()
296 Status GetCompileTimeConstInputs(const OpKernel* op_kernel, in GetCompileTimeConstInputs()
|
| D | xla_op_kernel.h | 307 const OpKernel& op_kernel() const { return context_->op_kernel(); } in op_kernel() function
|
| D | graph_compiler.cc | 140 std::unique_ptr<OpKernel> op_kernel(op_kernel_raw); in Compile() local
|
| D | xla_op_registry.h | 199 static Status CompileTimeConstantInputs(const OpKernel& op_kernel, in CompileTimeConstantInputs()
|
| D | xla_op_registry.cc | 389 const NodeDef& node_def, const OpKernel* op_kernel, const OpDef* op_def, in CompileTimeConstantInputs()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device.cc | 258 string PluggableDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
265 void PluggableDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
304 void PluggableDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | op_kernel_runner.cc | 123 std::unique_ptr<OpKernel> op_kernel; in Create() local
132 std::unique_ptr<tensorflow::OpKernel> op_kernel) in OpKernelRunner()
|
| D | op_kernel_runner.h | 86 tensorflow::OpKernel* op_kernel() const { return op_kernel_.get(); } in op_kernel() function
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_tfconv_op.h | 64 static void ConvertMklToTf(OpKernel* op_kernel, OpKernelContext* context, in ConvertMklToTf()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 71 std::unique_ptr<OpKernel> op_kernel( in EvaluateNode() local
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | list_flex_ops.cc | 39 Json::Value op_kernel(Json::arrayValue); in OpListToJSONString() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 534 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
559 bool ShouldLogInputsAndOutputs(OpKernel* op_kernel) { in ShouldLogInputsAndOutputs()
578 void BaseGPUDevice::LogInputs(OpKernel* op_kernel, OpKernelContext* context) { in LogInputs()
597 void BaseGPUDevice::LogOutputs(OpKernel* op_kernel, OpKernelContext* context) { in LogOutputs()
623 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
711 void BaseGPUDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl_1.cc | 41 Status ValidateUnsortedSegmentReduction(OpKernel* op_kernel, in ValidateUnsortedSegmentReduction()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 438 void XlaDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
445 void XlaDevice::ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op_test.cc | 107 Status CreateMapDefunContext(OpKernel* const op_kernel, in CreateMapDefunContext()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | dataset_test_base.cc | 299 const NodeDef& node_def, std::unique_ptr<OpKernel>* op_kernel) { in CreateOpKernel()
418 Status DatasetOpsTestBase::RunOpKernel(OpKernel* op_kernel, in RunOpKernel()
917 const DatasetParams& dataset_params, std::unique_ptr<OpKernel>* op_kernel) { in MakeGetOptionsOpKernel()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 577 auto op_kernel = absl::WrapUnique(op_kernel_raw); in LegalizeOp() local
|