| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.h | 85 virtual void Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
90 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| D | renamed_device.h | 132 void Compute(OpKernel* op_kernel, OpKernelContext* context) override { in Compute()
136 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| D | executor.cc | 818 const OpKernel* op_kernel = item->kernel; in SetAllocAttrs() local
1821 OpKernel* op_kernel = item.kernel; in Process() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl_1.cc | 41 void UnsortedSegmentReductionValidation(OpKernel* op_kernel, in UnsortedSegmentReductionValidation()
57 bool UnsortedSegmentReductionDoValidation(OpKernel* op_kernel, in UnsortedSegmentReductionDoValidation()
|
| D | mkl_tfconv_op.h | 71 static void ConvertMklToTf(OpKernel* op_kernel, OpKernelContext* context, in ConvertMklToTf()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 250 const OpKernel& op_kernel() const { return context_->op_kernel(); } in op_kernel() function
|
| D | const_analysis.cc | 88 Status GetCompileTimeConstInputs(const NodeDef& node, const OpKernel* op_kernel, in GetCompileTimeConstInputs()
258 Status GetCompileTimeConstInputs(const OpKernel* op_kernel, in GetCompileTimeConstInputs()
|
| D | xla_compilation_device.cc | 86 void XlaCompilationDevice::Compute(OpKernel* op_kernel, in Compute()
|
| D | graph_compiler.cc | 138 std::unique_ptr<OpKernel> op_kernel(op_kernel_raw); in Compile() local
|
| D | xla_op_registry.h | 195 static Status CompileTimeConstantInputs(const OpKernel& op_kernel, in CompileTimeConstantInputs()
|
| D | xla_op_registry.cc | 369 const NodeDef& node_def, const OpKernel* op_kernel, const OpDef* op_def, in CompileTimeConstantInputs()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 95 const OpKernel* op_kernel, int64 argument_idx, in MustArgumentBeConstant()
110 const OpKernel* op_kernel, int64 argument_idx, in ShouldArgumentBeConstant()
|
| D | xla_device.cc | 408 void XlaDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
415 void XlaDevice::ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.cc | 28 void SYCLDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 71 std::unique_ptr<OpKernel> op_kernel( in EvaluateNode() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 885 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
906 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
959 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
988 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
1000 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
1041 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
1051 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
|
| D | op_kernel.h | 632 OpKernel* op_kernel = nullptr; member
748 const OpKernel& op_kernel() const { return *params_->op_kernel; } in op_kernel() function
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op_test.cc | 107 Status CreateMapDefunContext(OpKernel* const op_kernel, in CreateMapDefunContext()
|
| D | single_threaded_executor.cc | 167 OpKernel* op_kernel = kernel_state.kernel; in Initialize() local
|
| D | dataset_test_base.cc | 305 const NodeDef& node_def, std::unique_ptr<OpKernel>* op_kernel) { in CreateOpKernel()
410 Status DatasetOpsTestBase::RunOpKernel(OpKernel* op_kernel, in RunOpKernel()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 472 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
479 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
558 void BaseGPUDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
|