| /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 | 602 OpKernel* op_kernel = nullptr; member
721 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 | xla_op_kernel.h | 262 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()
287 Status GetCompileTimeConstInputs(const OpKernel* op_kernel, in GetCompileTimeConstInputs()
|
| D | graph_compiler.cc | 141 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/ |
| 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 | executor.cc | 556 OpKernel* op_kernel = item.kernel; in ProcessSync() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_impl_1.cc | 43 void UnsortedSegmentReductionValidation(OpKernel* op_kernel, in UnsortedSegmentReductionValidation()
59 bool UnsortedSegmentReductionDoValidation(OpKernel* op_kernel, in UnsortedSegmentReductionDoValidation()
|
| /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/compiler/jit/ |
| 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/kernels/data/ |
| D | map_defun_op_test.cc | 107 Status CreateMapDefunContext(OpKernel* const op_kernel, in CreateMapDefunContext()
|
| D | single_threaded_executor.cc | 212 OpKernel* op_kernel = kernel_state.kernel; in Initialize() local
|
| D | dataset_test_base.cc | 304 const NodeDef& node_def, std::unique_ptr<OpKernel>* op_kernel) { in CreateOpKernel()
423 Status DatasetOpsTestBase::RunOpKernel(OpKernel* op_kernel, in RunOpKernel()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 444 auto op_kernel = absl::WrapUnique(op_kernel_raw); in LegalizeOp() local
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 531 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
538 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
615 void BaseGPUDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
|