| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device.h | 88 virtual void Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
93 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| D | renamed_device.h | 99 void Compute(OpKernel* op_kernel, OpKernelContext* context) override { in Compute()
103 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
|
| D | executor.cc | 768 const OpKernel* op_kernel = item->kernel; in SetAllocAttrs() local
1730 OpKernel* op_kernel = item.kernel; in Process() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_op_kernel.h | 234 const OpKernel& op_kernel() const { return context_->op_kernel(); } in op_kernel() function
|
| D | xla_compilation_device.cc | 90 void XlaCompilationDevice::Compute(OpKernel* op_kernel, in Compute()
|
| D | graph_compiler.cc | 135 std::unique_ptr<OpKernel> op_kernel(op_kernel_raw); in Compile() local
|
| D | xla_op_registry.h | 160 static Status CompileTimeConstantInputs(const OpKernel& op_kernel, in CompileTimeConstantInputs()
|
| D | xla_op_registry.cc | 361 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 | 91 Status XlaCompileOnDemandOp::MustArgumentBeConstant(const OpKernel* op_kernel, in MustArgumentBeConstant()
104 Status XlaCompileOnDemandOp::ShouldArgumentBeConstant(const OpKernel* op_kernel, in ShouldArgumentBeConstant()
|
| D | xla_device.cc | 373 void XlaDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
379 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/framework/ |
| D | op_kernel_test.cc | 753 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
774 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
827 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
856 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
868 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
909 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
919 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
|
| D | op_kernel.h | 584 OpKernel* op_kernel = nullptr; member
695 const OpKernel& op_kernel() const { return *params_->op_kernel; } in op_kernel() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_tfconv_op.h | 61 static void ConvertMklToTf(OpKernel* op_kernel, OpKernelContext* context, in ConvertMklToTf()
|
| D | segment_reduction_ops.cc | 436 static void UnsortedSegmentReductionValidation(OpKernel* op_kernel, in UnsortedSegmentReductionValidation()
452 static bool UnsortedSegmentReductionDoValidation(OpKernel* op_kernel, in UnsortedSegmentReductionDoValidation()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 90 const NodeDef& node_def, std::unique_ptr<OpKernel>* op_kernel) { in CreateOpKernel()
172 Status DatasetOpsTestBase::RunOpKernel(OpKernel* op_kernel, in RunOpKernel()
|
| D | concatenate_dataset_op_test.cc | 50 std::unique_ptr<OpKernel> *op_kernel) { in CreateConcatenateDatasetKernel()
60 OpKernel *const op_kernel, in CreateConcatenateDatasetContext()
|
| D | zip_dataset_op_test.cc | 54 std::unique_ptr<OpKernel> *op_kernel) { in CreateZipDatasetKernel()
70 OpKernel *const op_kernel, in CreateZipDatasetContext()
|
| D | take_dataset_op_test.cc | 40 std::unique_ptr<OpKernel> *op_kernel) { in CreateTakeDatasetKernel()
50 OpKernel *op_kernel, gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateTakeDatasetContext()
|
| D | repeat_dataset_op_test.cc | 43 std::unique_ptr<OpKernel> *op_kernel) { in CreateRepeatDatasetKernel()
53 OpKernel *op_kernel, gtl::InlinedVector<TensorValue, 4> *const inputs, in CreateRepeatDatasetContext()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 41 DataType tvalues, std::unique_ptr<OpKernel> *op_kernel) { in CreateSparseTensorSliceDatasetKernel()
51 OpKernel *const op_kernel, gtl::InlinedVector<TensorValue, 4> *inputs, in CreateSparseTensorSliceDatasetContext()
|
| D | single_threaded_executor.cc | 126 OpKernel* op_kernel = kernel_state.kernel; in Initialize() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 92 std::unique_ptr<OpKernel> op_kernel( in EvaluateNode() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 489 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
507 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
514 void BaseGPUDevice::ComputeHelper(OpKernel* op_kernel, in ComputeHelper()
615 void BaseGPUDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
|