Home
last modified time | relevance | path

Searched defs:op_kernel (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dthreadpool_device.cc166 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()
H A Drenamed_device.h123 void Compute(OpKernel* op_kernel, OpKernelContext* context) override { in Compute()
127 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
H A Dnode_file_writer.cc127 Status NodeFileWriter::RecordNodeExecution(OpKernel* op_kernel, in RecordNodeExecution()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Doss_proxy_executor.cpp19 OSSOpKernel& op_kernel) { in prefill_stack_with_static_arguments()
231 OSSOpKernel& op_kernel) { in get_input_info_from_serialized()
248 OSSOpKernel& op_kernel) { in get_output_info_from_serialized()
349 OSSOpKernel op_kernel(target, op_handle); in OSSProxyExecutor() local
366 OSSOpKernel& op_kernel = op_kernels_[extern_node_index]; in call_function() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel_test.cc710 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
H A Ddevice.h87 virtual void Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
92 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device.cc258 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_compilation_device.cc91 OpKernel* op_kernel, XlaContext& context) { in AttachLocationToMetadata()
101 void XlaCompilationDevice::Compute(OpKernel* op_kernel, in Compute()
H A Dconst_analysis.cc89 Status GetCompileTimeConstInputs(const NodeDef& node, const OpKernel* op_kernel, in GetCompileTimeConstInputs()
296 Status GetCompileTimeConstInputs(const OpKernel* op_kernel, in GetCompileTimeConstInputs()
H A Dxla_op_kernel.h312 const OpKernel& op_kernel() const { return context_->op_kernel(); } in op_kernel() function
H A Dgraph_compiler.cc141 std::unique_ptr<OpKernel> op_kernel(op_kernel_raw); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc557 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/
H A Dop_kernel_runner.cc126 std::unique_ptr<OpKernel> op_kernel; in Create() local
135 std::unique_ptr<tensorflow::OpKernel> op_kernel) in OpKernelRunner()
H A Dop_kernel_runner.h67 tensorflow::OpKernel* op_kernel() const { return op_kernel_.get(); } in op_kernel() function
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dop_kernel_runner.cc123 std::unique_ptr<OpKernel> op_kernel; in Create() local
132 std::unique_ptr<tensorflow::OpKernel> op_kernel) in OpKernelRunner()
H A Dop_kernel_runner.h86 tensorflow::OpKernel* op_kernel() const { return op_kernel_.get(); } in op_kernel() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_device.cc438 void XlaDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
445 void XlaDevice::ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Ddataset_test_base.cc299 const NodeDef& node_def, std::unique_ptr<OpKernel>* op_kernel) { in CreateOpKernel()
428 Status DatasetOpsTestBase::RunOpKernel(OpKernel* op_kernel, in RunOpKernel()
927 const DatasetParams& dataset_params, std::unique_ptr<OpKernel>* op_kernel) { in MakeGetOptionsOpKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_tfconv_op.h64 static void ConvertMklToTf(OpKernel* op_kernel, OpKernelContext* context, in ConvertMklToTf()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Devaluation_utils.cc71 std::unique_ptr<OpKernel> op_kernel( in EvaluateNode() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dlist_flex_ops.cc39 Json::Value op_kernel(Json::arrayValue); in OpListToJSONString() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/utils/
H A Deval_utils.cc114 std::unique_ptr<tensorflow::OpKernel> op_kernel = tensorflow::CreateOpKernel( in EvaluateOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsegment_reduction_ops_impl_1.cc41 Status ValidateUnsortedSegmentReduction(OpKernel* op_kernel, in ValidateUnsortedSegmentReduction()
H A Dops_testutil.cc221 OpKernel* OpsTestBase::op_kernel() { return kernel_.get(); } in op_kernel() function in tensorflow::OpsTestBase
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dmap_defun_op_test.cc107 Status CreateMapDefunContext(OpKernel* const op_kernel, in CreateMapDefunContext()

12