| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device.cc | 166 bool ShouldLogInputsAndOutputs(OpKernel* op_kernel) { in ShouldLogInputsAndOutputs()
177 void ThreadPoolDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
199 void ThreadPoolDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
216 void ThreadPoolDevice::LogInputs(OpKernel* op_kernel, in LogInputs()
231 void ThreadPoolDevice::LogOutputs(OpKernel* op_kernel, in LogOutputs()
|
| D | renamed_device.h | 123 void Compute(OpKernel* op_kernel, OpKernelContext* context) override { in Compute()
127 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| D | node_file_writer.cc | 127 Status NodeFileWriter::RecordNodeExecution(OpKernel* op_kernel, in RecordNodeExecution()
|
| D | graph_view.cc | 447 const OpKernel* op_kernel = item->kernel; in SetAllocAttrs() 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 | 710 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
731 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
784 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
821 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
833 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
867 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
877 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
|
| /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 | 312 const OpKernel& op_kernel() const { return context_->op_kernel(); } in op_kernel() function
|
| D | graph_compiler.cc | 141 std::unique_ptr<OpKernel> op_kernel(op_kernel_raw); in Compile() local
|
| /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/tfrt/fallback/ |
| D | op_kernel_runner.cc | 126 std::unique_ptr<OpKernel> op_kernel; in Create() local
135 std::unique_ptr<tensorflow::OpKernel> op_kernel) in OpKernelRunner()
|
| D | op_kernel_runner.h | 67 tensorflow::OpKernel* op_kernel() const { return op_kernel_.get(); } in op_kernel() function
|
| /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
|
| D | kernel_fallback_op_handler.cc | 136 auto* op_kernel = op_entry.op_kernel_runner->op_kernel(); in GetResultDevice() local
|
| /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/transforms/utils/ |
| D | eval_utils.cc | 114 std::unique_ptr<tensorflow::OpKernel> op_kernel = tensorflow::CreateOpKernel( in EvaluateOperation() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl_1.cc | 41 Status ValidateUnsortedSegmentReduction(OpKernel* op_kernel, in ValidateUnsortedSegmentReduction()
|
| D | ops_testutil.cc | 221 OpKernel* OpsTestBase::op_kernel() { return kernel_.get(); } in op_kernel() function in tensorflow::OpsTestBase
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 557 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
582 bool ShouldLogInputsAndOutputs(OpKernel* op_kernel) { in ShouldLogInputsAndOutputs()
601 void BaseGPUDevice::LogInputs(OpKernel* op_kernel, OpKernelContext* context) { in LogInputs()
620 void BaseGPUDevice::LogOutputs(OpKernel* op_kernel, OpKernelContext* context) { in LogOutputs()
646 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
741 void BaseGPUDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
|
| /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()
|