| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device.cc | 166 bool ShouldLogInputsAndOutputs(OpKernel* op_kernel) { in ShouldLogInputsAndOutputs() argument
173 return ops_to_log.count(op_kernel->type_string()); in ShouldLogInputsAndOutputs()
177 void ThreadPoolDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute() argument
178 bool should_log_inputs_and_outputs = ShouldLogInputsAndOutputs(op_kernel); in Compute()
181 LogInputs(op_kernel, context); in Compute()
184 op_kernel->Compute(context); in Compute()
187 Status s = node_file_writer_->RecordNodeExecution(op_kernel, context); in Compute()
195 LogOutputs(op_kernel, context); in Compute()
199 void ThreadPoolDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync() argument
202 bool should_log_inputs_and_outputs = ShouldLogInputsAndOutputs(op_kernel); in ComputeAsync()
[all …]
|
| D | threadpool_device.h | 48 void Compute(OpKernel* op_kernel, OpKernelContext* context) override;
49 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context,
53 void LogInputs(OpKernel* op_kernel, OpKernelContext* context);
54 void LogOutputs(OpKernel* op_kernel, OpKernelContext* context);
|
| D | renamed_device.h | 123 void Compute(OpKernel* op_kernel, OpKernelContext* context) override { in Compute() argument
124 underlying_device_->Compute(op_kernel, context); in Compute()
127 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() argument
129 underlying_device_->ComputeAsync(op_kernel, context, std::move(done)); in ComputeAsync()
|
| D | node_file_writer.cc | 127 Status NodeFileWriter::RecordNodeExecution(OpKernel* op_kernel, in RecordNodeExecution() argument
129 if (kOpsToSkipWriting->count(op_kernel->type_string())) { in RecordNodeExecution()
134 def.set_op(op_kernel->def().op()); in RecordNodeExecution()
135 *def.mutable_attr() = op_kernel->def().attr(); in RecordNodeExecution()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 91 OpKernel* op_kernel, XlaContext& context) { in AttachLocationToMetadata() argument
93 context.StackTraceForNodeName(op_kernel->def().name())) { in AttachLocationToMetadata()
101 void XlaCompilationDevice::Compute(OpKernel* op_kernel, in Compute() argument
104 << FormatNodeDefForError(op_kernel->def()); in Compute()
108 metadata.set_op_type(op_kernel->type_string()); in Compute()
109 metadata.set_op_name(op_kernel->name()); in Compute()
110 AttachLocationToMetadata(metadata, op_kernel, xla_context); in Compute()
114 ParseShardingFromDevice(op_kernel->def(), std::numeric_limits<int>::max(), in Compute()
121 GetFrontendAttributesFromAttrSlice(AttrSlice(op_kernel->def())); in Compute()
138 op_kernel->Compute(context); in Compute()
|
| D | const_analysis.cc | 89 Status GetCompileTimeConstInputs(const NodeDef& node, const OpKernel* op_kernel, in GetCompileTimeConstInputs() argument
93 DCHECK(op_def != nullptr || op_kernel != nullptr); in GetCompileTimeConstInputs()
169 return XlaOpRegistry::CompileTimeConstantInputs(*op_kernel, in GetCompileTimeConstInputs()
296 Status GetCompileTimeConstInputs(const OpKernel* op_kernel, in GetCompileTimeConstInputs() argument
299 return GetCompileTimeConstInputs(op_kernel->def(), op_kernel, in GetCompileTimeConstInputs()
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/ |
| D | oss_proxy_executor.cpp | 19 OSSOpKernel& op_kernel) { in prefill_stack_with_static_arguments() argument
20 auto& stack = op_kernel.stack_; in prefill_stack_with_static_arguments()
21 auto& dynamic_args = op_kernel.dynamic_args_; in prefill_stack_with_static_arguments()
92 << op_kernel.target_ << " argument #" << index in prefill_stack_with_static_arguments()
217 index, inner_type, serialized_arg, op_kernel); in prefill_stack_with_static_arguments()
231 OSSOpKernel& op_kernel) { in get_input_info_from_serialized() argument
238 index++, schema_arg.real_type(), arg, op_kernel); in get_input_info_from_serialized()
248 OSSOpKernel& op_kernel) { in get_output_info_from_serialized() argument
249 std::vector<OSSDynamicArg>& outputs = op_kernel.outputs_; in get_output_info_from_serialized()
349 OSSOpKernel op_kernel(target, op_handle); in OSSProxyExecutor() local
[all …]
|
| D | oss_proxy_executor.h | 87 OSSOpKernel& op_kernel);
92 OSSOpKernel& op_kernel);
97 OSSOpKernel& op_kernel);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_common.cc | 39 const string& op = ctx->op_kernel().type_string(); in SetComputeError()
41 DataTypeIsInteger(ctx->op_kernel().input_type(0))) { in SetComputeError()
44 DataTypeIsInteger(ctx->op_kernel().input_type(0)) && in SetComputeError()
45 DataTypeIsSigned(ctx->op_kernel().input_type(1))) { in SetComputeError()
62 TryGetNodeAttr(ctx->op_kernel().def(), "incompatible_shape_error", in BinaryOpState()
65 const string& op = ctx->op_kernel().type_string(); in BinaryOpState()
|
| D | scoped_allocator_ops.cc | 61 VLOG(1) << "_ScopedAllocatorOp " << context->op_kernel().name() in Compute()
144 context->op_kernel().requested_input(i), " to node ", in Compute()
145 context->op_kernel().name(), " input bounds = [", input_lb, ", ", in Compute()
152 context->op_kernel().requested_input(i), " to node ", in Compute()
153 context->op_kernel().name(), " input bounds = [", input_lb, ", ", in Compute()
218 " to node ", context->op_kernel().name())); in Compute()
224 " to node ", context->op_kernel().name())); in Compute()
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device.cc | 258 string PluggableDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString() argument
260 return strings::StrCat(op_kernel.name(), " op ", op_kernel.type_string(), in ComputeOpKernelDebugString()
265 void PluggableDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute() argument
277 << ComputeOpKernelDebugString(*op_kernel, stream_id); in Compute()
280 op_kernel->Compute(context); in Compute()
286 << ComputeOpKernelDebugString(*op_kernel, stream_id); in Compute()
290 << ComputeOpKernelDebugString(*op_kernel, stream_id); in Compute()
295 << ComputeOpKernelDebugString(*op_kernel, stream_id); in Compute()
304 void PluggableDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync() argument
314 VLOG(1) << "PluggableDevice::ComputeAsync " << op_kernel->name() << " op " in ComputeAsync()
[all …]
|
| D | pluggable_device.h | 59 void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context,
62 void Compute(OpKernel* op_kernel, OpKernelContext* context) override;
103 std::string ComputeOpKernelDebugString(const OpKernel& op_kernel,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 71 std::unique_ptr<OpKernel> op_kernel( in EvaluateNode() local
79 params.op_kernel = op_kernel.get(); in EvaluateNode()
83 const int num_outputs = op_kernel->num_outputs(); in EvaluateNode()
92 op_kernel->Compute(&op_context); in EvaluateNode()
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | eval_utils.cc | 114 std::unique_ptr<tensorflow::OpKernel> op_kernel = tensorflow::CreateOpKernel( in EvaluateOperation() local
126 params.op_kernel = op_kernel.get(); in EvaluateOperation()
130 op_kernel->num_outputs()); in EvaluateOperation()
136 op_kernel->Compute(&op_context); in EvaluateOperation()
144 for (int i = 0; i < op_kernel->num_outputs(); ++i) { in EvaluateOperation()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel_test.cc | 371 params.op_kernel = op.get(); in TEST_F()
455 params.op_kernel = op.get(); in TEST_F()
627 params.op_kernel = op.get(); in TEST_F()
710 std::unique_ptr<OpKernel> op_kernel = in TEST_F() local
713 auto* get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()
717 op_kernel = ExpectSuccess("GetAttrStringList", DEVICE_CPU, in TEST_F()
719 get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()
731 std::unique_ptr<OpKernel> op_kernel = ExpectSuccess( in TEST_F() local
734 auto* get_attr_kernel = static_cast<GetAttrKernel*>(op_kernel.get()); in TEST_F()
739 op_kernel = ExpectSuccess( in TEST_F()
[all …]
|
| D | device.h | 87 virtual void Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute() argument
88 op_kernel->Compute(context); in Compute()
92 virtual void ComputeAsync(AsyncOpKernel* op_kernel, OpKernelContext* context, in ComputeAsync() argument
94 op_kernel->ComputeAsync(context, std::move(done)); in ComputeAsync()
|
| D | op_kernel.cc | 120 bool ShouldLogNodeDef(OpKernel* op_kernel) { in ShouldLogNodeDef() argument
123 return ops_to_log_nodedefs.count(op_kernel->type_string()); in ShouldLogNodeDef()
340 params, static_cast<int>(params->op_kernel->output_types().size())) {} in OpKernelContext()
434 CHECK_LT(index, num_inputs()) << " name: " << op_kernel().name(); in input()
672 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in input_list()
680 TF_RETURN_IF_ERROR(params_->op_kernel->InputRange(name, &start, &stop)); in mutable_input_list()
687 TF_RETURN_IF_ERROR(params_->op_kernel->OutputRange(name, &start, &stop)); in output_list()
702 " kernel=", params_->op_kernel->name()); in allocate_output()
707 " kernel=", params_->op_kernel->name()); in allocate_output()
725 TF_RETURN_IF_ERROR(params_->op_kernel->OutputRange(name, &start, &stop)); in allocate_output()
[all …]
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | op_kernel_runner.cc | 123 std::unique_ptr<OpKernel> op_kernel; in Create() local
125 std::move(node_def), &op_kernel)); in Create()
126 return OpKernelRunner(device, function_library_runtime, std::move(op_kernel)); in Create()
132 std::unique_ptr<tensorflow::OpKernel> op_kernel) in OpKernelRunner() argument
136 op_kernel_(std::move(op_kernel)), in OpKernelRunner()
179 DCHECK_EQ(it->second->op_kernel()->name(), op_name); in GetOrCreate()
188 DCHECK_EQ(it->second->op_kernel()->name(), op_name); in GetOrCreate()
|
| D | kernel_fallback_op_handler.cc | 136 auto* op_kernel = op_entry.op_kernel_runner->op_kernel(); in GetResultDevice() local
137 DCHECK(index < op_kernel->num_outputs()); in GetResultDevice()
140 if (op_kernel->output_memory_types()[index] == MemoryType::HOST_MEMORY && in GetResultDevice()
141 op_kernel->output_type(index) != DT_RESOURCE) { in GetResultDevice()
201 DCHECK_EQ(it->second->op_kernel()->name(), op_name); in GetOrCreate()
210 DCHECK_EQ(it->second->op_kernel()->name(), op_name); in GetOrCreate()
|
| /external/tensorflow/tensorflow/lite/kernels/shim/ |
| D | BUILD | 14 "op_kernel.h",
28 ":op_kernel",
135 name = "op_kernel",
136 hdrs = ["op_kernel.h"],
155 ":op_kernel",
179 ":op_kernel",
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | list_flex_ops.cc | 39 Json::Value op_kernel(Json::arrayValue); in OpListToJSONString() local
40 op_kernel.append(Json::Value(op.op_name)); in OpListToJSONString()
41 op_kernel.append(Json::Value(op.kernel_name)); in OpListToJSONString()
42 result.append(op_kernel); in OpListToJSONString()
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | op_kernel_runner.cc | 126 std::unique_ptr<OpKernel> op_kernel; in Create() local
128 std::move(node_def), &op_kernel)); in Create()
129 return OpKernelRunner(device, function_library_runtime, std::move(op_kernel)); in Create()
135 std::unique_ptr<tensorflow::OpKernel> op_kernel) in OpKernelRunner() argument
139 op_kernel_(std::move(op_kernel)), in OpKernelRunner()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 132 operation_ = context->op_kernel().name() + "/EigenAllocator"; in Reinitialize()
557 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString() argument
559 return strings::StrCat(op_kernel.name(), " op ", op_kernel.type_string(), in ComputeOpKernelDebugString()
582 bool ShouldLogInputsAndOutputs(OpKernel* op_kernel) { in ShouldLogInputsAndOutputs() argument
585 return ops_to_log.count(op_kernel->type_string()); in ShouldLogInputsAndOutputs()
601 void BaseGPUDevice::LogInputs(OpKernel* op_kernel, OpKernelContext* context) { in LogInputs() argument
602 LOG(INFO) << "Inputs for " << op_kernel->name() << " (total " in LogInputs()
620 void BaseGPUDevice::LogOutputs(OpKernel* op_kernel, OpKernelContext* context) { in LogOutputs() argument
622 LOG(INFO) << op_kernel->name() in LogOutputs()
627 LOG(INFO) << "Outputs for " << op_kernel->name() << " (total " in LogOutputs()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/batching_util/ |
| D | batch_resource_base.cc | 260 context->op_kernel().name()); in RegisterInput()
262 context->op_kernel().name()); in RegisterInput()
266 context->op_kernel().name()); in RegisterInput()
269 context->op_kernel().name()); in RegisterInput()
272 context->op_kernel().name()); in RegisterInput()
276 GetModelName(context), context->op_kernel().name()); in RegisterInput()
279 context->op_kernel().name()); in RegisterInput()
282 GetModelName(context), context->op_kernel().name()); in RegisterInput()
288 context->op_kernel().name()); in RegisterInput()
436 context->op_kernel().name()); in ConcatInputTensors()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_tfconv_op.h | 64 static void ConvertMklToTf(OpKernel* op_kernel, OpKernelContext* context, in ConvertMklToTf() argument
83 DataType input_data_type = op_kernel->input_type(input_number); in ConvertMklToTf()
84 DataType output_data_type = op_kernel->output_type(input_number); in ConvertMklToTf()
|