Home
last modified time | relevance | path

Searched refs:op_info (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc82 string GetDataFormat(const OpInfo& op_info) { in GetDataFormat() argument
84 if (op_info.attr().find("data_format") != op_info.attr().end()) { in GetDataFormat()
85 data_format = op_info.attr().at("data_format").s(); in GetDataFormat()
90 string GetFilterFormat(const OpInfo& op_info) { in GetFilterFormat() argument
92 if (op_info.attr().find("filter_format") != op_info.attr().end()) { in GetFilterFormat()
93 filter_format = op_info.attr().at("filter_format").s(); in GetFilterFormat()
98 Padding GetPadding(const OpInfo& op_info) { in GetPadding() argument
99 if (op_info.attr().find("padding") != op_info.attr().end() && in GetPadding()
100 op_info.attr().at("padding").s() == "VALID") { in GetPadding()
106 bool IsTraining(const OpInfo& op_info) { in IsTraining() argument
[all …]
Dop_level_cost_estimator_test.cc57 void DescribeMatrix(int rows, int columns, OpInfo* op_info) { in DescribeMatrix() argument
58 auto input = op_info->add_inputs(); in DescribeMatrix()
67 void SetCpuDevice(OpInfo* op_info) { in SetCpuDevice() argument
68 auto device = op_info->mutable_device(); in SetCpuDevice()
78 SetCpuDevice(&op_context.op_info); in DescribeMatMul()
79 op_context.op_info.set_op("MatMul"); in DescribeMatMul()
81 DescribeMatrix(m, l, &op_context.op_info); in DescribeMatMul()
82 DescribeMatrix(k, n, &op_context.op_info); in DescribeMatMul()
89 OpInfo* op_info) { in DescribeArbitraryRankInput() argument
90 auto input = op_info->add_inputs(); in DescribeArbitraryRankInput()
[all …]
Dop_level_cost_estimator.h48 Costs PredictOpCountBasedCost(double operations, const OpInfo& op_info) const;
55 const OpInfo& op_info) const;
78 int64 CountConv2DOperations(const OpInfo& op_info,
80 int64 CountConv2DOperations(const OpInfo& op_info,
83 int64 CountMatMulOperations(const OpInfo& op_info,
85 int64 CountMatMulOperations(const OpInfo& op_info, MatMulDimensions* mat_mul,
87 int64 CountBatchMatMulOperations(const OpInfo& op_info,
90 const OpInfo& op_info, ConvolutionDimensions* returned_conv_dims,
93 const OpInfo& op_info, ConvolutionDimensions* returned_conv_dims,
106 int64 CalculateLargestInputCount(const OpInfo& op_info,
[all …]
Dutils.cc82 OpInfo* op_info) { in ExtractExtraProperties() argument
103 if (i >= op_info->inputs_size()) { in ExtractExtraProperties()
105 << op_info->DebugString() in ExtractExtraProperties()
111 if (input_node->op() == "Const" && i < op_info->inputs_size()) { in ExtractExtraProperties()
120 OpInfo::TensorProperties* input = op_info->mutable_inputs(i); in ExtractExtraProperties()
144 (*op_info->mutable_attr())[attr_key] = attr; in ExtractExtraProperties()
155 (*op_info->mutable_attr())[new_key] = attr; in ExtractExtraProperties()
266 OpInfo op_info; in BuildOpInfoWithoutDevice() local
267 op_info.set_op(node.op()); in BuildOpInfoWithoutDevice()
268 *op_info.mutable_attr() = node.attr(); in BuildOpInfoWithoutDevice()
[all …]
Dvirtual_scheduler.cc635 auto& op_info = op_context.op_info; in GetCurrNode() local
636 op_info.set_op(node->op()); in GetCurrNode()
637 *op_info.mutable_attr() = node->attr(); in GetCurrNode()
639 *op_info.add_inputs() = input; in GetCurrNode()
642 *op_info.add_outputs() = output; in GetCurrNode()
644 op_info.mutable_device()->Swap(&device); in GetCurrNode()
767 string node_description = GetOpDescription(op_context.op_info); in MarkCurrNodeExecuted()
Dop_context.h30 OpInfo op_info; member
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc81 auto op_info = node_ptr_opinfo_map_.at(node); in GetKernelInvoke() local
82 return logging_op_resolver_->GetWrappedKernelInvoke(op_info.builtin_op_code, in GetKernelInvoke()
170 auto op_info = calibrator->GetOpInfo(node); in LoggingEval() local
172 for (int i : op_info.loggable_inputs) { in LoggingEval()
183 for (int i : op_info.loggable_outputs) { in LoggingEval()
288 OperatorInfo op_info; in BuildLoggingInterpreter() local
289 op_info.node_index = i; in BuildLoggingInterpreter()
292 op_info.builtin_op_code = operator_code->builtin_code(); in BuildLoggingInterpreter()
293 op_info.name = GetOpName(*operator_code); in BuildLoggingInterpreter()
294 op_info.is_custom_op = operator_code->custom_code() != nullptr; in BuildLoggingInterpreter()
[all …]
Dnode_info_delegate.cc52 auto op_info = node_index_opinfo_map_.at(node_index); in OnDelegatePrepareCalled() local
53 op_info.registration = reg; in OnDelegatePrepareCalled()
54 node_ptr_opinfo_map_->insert({node, op_info}); in OnDelegatePrepareCalled()
/external/capstone/bindings/java/capstone/
DArm.java135 public OpInfo(UnionOpInfo op_info) { in OpInfo() argument
136 usermode = op_info.usermode; in OpInfo()
137 vectorSize = op_info.vector_size; in OpInfo()
138 vectorData = op_info.vector_data; in OpInfo()
139 cpsMode = op_info.cps_mode; in OpInfo()
140 cpsFlag = op_info.cps_flag; in OpInfo()
141 cc = op_info.cc; in OpInfo()
142 updateFlags = (op_info.update_flags > 0); in OpInfo()
143 writeback = (op_info.writeback > 0); in OpInfo()
144 memBarrier = op_info.mem_barrier; in OpInfo()
[all …]
DPpc.java102 public OpInfo(UnionOpInfo op_info) { in OpInfo() argument
103 bc = op_info.bc; in OpInfo()
104 bh = op_info.bh; in OpInfo()
105 updateCr0 = (op_info.update_cr0 > 0); in OpInfo()
106 op = op_info.op; in OpInfo()
DArm64.java120 public OpInfo(UnionOpInfo op_info) { in OpInfo() argument
121 cc = op_info.cc; in OpInfo()
122 updateFlags = (op_info._update_flags > 0); in OpInfo()
123 writeback = (op_info._writeback > 0); in OpInfo()
124 op = op_info.op; in OpInfo()
DSparc.java86 public OpInfo(UnionOpInfo op_info) { in OpInfo() argument
87 cc = op_info.cc; in OpInfo()
88 hint = op_info.hint; in OpInfo()
89 op = op_info.op; in OpInfo()
DCapstone.java156 OpInfo op_info = null; in getOptInfo() local
162 op_info = new Arm.OpInfo((Arm.UnionOpInfo) detail.arch.arm); in getOptInfo()
167 op_info = new Arm64.OpInfo((Arm64.UnionOpInfo) detail.arch.arm64); in getOptInfo()
172 op_info = new Mips.OpInfo((Mips.UnionOpInfo) detail.arch.mips); in getOptInfo()
177 op_info = new X86.OpInfo((X86.UnionOpInfo) detail.arch.x86); in getOptInfo()
182 op_info = new Sparc.OpInfo((Sparc.UnionOpInfo) detail.arch.sparc); in getOptInfo()
187 op_info = new Systemz.OpInfo((Systemz.UnionOpInfo) detail.arch.sysz); in getOptInfo()
192 op_info = new Ppc.OpInfo((Ppc.UnionOpInfo) detail.arch.ppc); in getOptInfo()
197 op_info = new Xcore.OpInfo((Xcore.UnionOpInfo) detail.arch.xcore); in getOptInfo()
202 return op_info; in getOptInfo()
DSystemz.java86 public OpInfo(UnionOpInfo op_info) { in OpInfo() argument
87 cc = op_info.cc; in OpInfo()
88 op = op_info.op; in OpInfo()
DXcore.java79 public OpInfo(UnionOpInfo op_info) { in OpInfo() argument
80 op = op_info.op; in OpInfo()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dstatic_schedule.cc34 op_context.op_info.set_op(node.op()); in PredictExecutionTime()
35 *op_context.op_info.mutable_attr() = node.attr(); in PredictExecutionTime()
40 op_context.op_info.add_inputs()->Swap(&input); in PredictExecutionTime()
46 op_context.op_info.add_outputs()->Swap(&output); in PredictExecutionTime()
50 op_context.op_info.mutable_device()->Swap(&device); in PredictExecutionTime()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes.cc647 for (const QuantizedOpInfo& op_info : op_list) { in QuantizeNodes() local
648 if (ops_to_ignore.count(op_info.float_name) == 0) { in QuantizeNodes()
650 op_info.float_name); in QuantizeNodes()
651 op_map.insert({op_info.float_name, op_info}); in QuantizeNodes()
700 const QuantizedOpInfo& op_info = op_map[float_node.op()]; in QuantizeNodes() local
709 if (op_info.unquantized_inputs.count(i)) { in QuantizeNodes()
733 if (op_info.unquantized_inputs.count(i)) { in QuantizeNodes()
805 for (const string& attr_to_copy : op_info.attrs_to_copy) { in QuantizeNodes()
810 op_info.dtypes_to_set) { in QuantizeNodes()
816 if (op_info.unquantized_inputs.count(i)) { in QuantizeNodes()
[all …]
/external/mesa3d/src/compiler/nir/
Dnir_builder.h296 const nir_op_info *op_info = &nir_op_infos[op]; in nir_build_alu() local
314 unsigned num_components = op_info->output_size; in nir_build_alu()
316 for (unsigned i = 0; i < op_info->num_inputs; i++) { in nir_build_alu()
317 if (op_info->input_sizes[i] == 0) in nir_build_alu()
327 unsigned bit_size = nir_alu_type_get_type_size(op_info->output_type); in nir_build_alu()
329 for (unsigned i = 0; i < op_info->num_inputs; i++) { in nir_build_alu()
331 if (nir_alu_type_get_type_size(op_info->input_types[i]) == 0) { in nir_build_alu()
338 nir_alu_type_get_type_size(op_info->input_types[i])); in nir_build_alu()
350 for (unsigned i = 0; i < op_info->num_inputs; i++) { in nir_build_alu()
/external/tensorflow/tensorflow/compiler/jit/
Dresource_operation_safety_analysis.cc128 const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(n.type_string()); in XlaResourceOpKindForNode() local
129 if (op_info) { in XlaResourceOpKindForNode()
130 *out_resource_op_kind = op_info->kind(); in XlaResourceOpKindForNode()
Dmark_for_compilation_pass.cc159 const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(node.type_string()); in IsNonResourceVarResourceOp() local
160 return op_info && op_info->resource_kind() != XlaResourceKind::kVariable; in IsNonResourceVarResourceOp()
597 const XlaResourceOpInfo* op_info = in FindCompilationCandidates() local
600 op_info && op_info->resource_kind() != XlaResourceKind::kVariable; in FindCompilationCandidates()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc70 static OpWithBiasTensors op_info = {/* activation_input_index */ 0, in GetInfoForOpWithBiasTensor() local
74 return &op_info; in GetInfoForOpWithBiasTensor()
77 static OpWithBiasTensors op_info = {/* bias_input_index */ 0, in GetInfoForOpWithBiasTensor() local
81 return &op_info; in GetInfoForOpWithBiasTensor()
/external/iproute2/devlink/
Ddevlink.c3095 struct dpipe_op_info *op_info = &action->info; in pr_out_dpipe_action() local
3101 dpipe_header_id2s(ctx, op_info->header_id, in pr_out_dpipe_action()
3102 op_info->header_global)); in pr_out_dpipe_action()
3104 dpipe_field_id2s(ctx, op_info->header_id, in pr_out_dpipe_action()
3105 op_info->field_id, in pr_out_dpipe_action()
3106 op_info->header_global)); in pr_out_dpipe_action()
3107 mapping = dpipe_mapping_get(ctx, op_info->header_id, in pr_out_dpipe_action()
3108 op_info->field_id, in pr_out_dpipe_action()
3109 op_info->header_global); in pr_out_dpipe_action()
3177 struct dpipe_op_info *op_info = &match->info; in pr_out_dpipe_match() local
[all …]
/external/llvm/tools/llvm-objdump/
DMachODump.cpp1814 struct LLVMOpInfo1 *op_info = (struct LLVMOpInfo1 *)TagBuf; in SymbolizerGetOpInfo() local
1815 uint64_t value = op_info->Value; in SymbolizerGetOpInfo()
1818 memset((void *)op_info, '\0', sizeof(struct LLVMOpInfo1)); in SymbolizerGetOpInfo()
1819 op_info->Value = value; in SymbolizerGetOpInfo()
1890 op_info->AddSymbol.Present = 1; in SymbolizerGetOpInfo()
1891 op_info->AddSymbol.Name = name; in SymbolizerGetOpInfo()
1901 op_info->AddSymbol.Present = 1; in SymbolizerGetOpInfo()
1903 op_info->AddSymbol.Name = add; in SymbolizerGetOpInfo()
1905 op_info->AddSymbol.Value = r_value; in SymbolizerGetOpInfo()
1906 op_info->SubtractSymbol.Present = 1; in SymbolizerGetOpInfo()
[all …]
/external/mesa3d/src/gallium/drivers/r600/sb/
Dsb_pass.h229 struct op_info { struct
232 op_info() : top_bb(), bottom_bb() {} in op_info() argument
235 typedef std::map<node*, op_info> op_info_map;
292 void bu_find_best_bb(node *n, op_info &oi);
/external/swiftshader/third_party/llvm-7.0/llvm/tools/llvm-objdump/
DMachODump.cpp2217 struct LLVMOpInfo1 *op_info = (struct LLVMOpInfo1 *)TagBuf; in SymbolizerGetOpInfo() local
2218 uint64_t value = op_info->Value; in SymbolizerGetOpInfo()
2221 memset((void *)op_info, '\0', sizeof(struct LLVMOpInfo1)); in SymbolizerGetOpInfo()
2222 op_info->Value = value; in SymbolizerGetOpInfo()
2288 op_info->AddSymbol.Present = 1; in SymbolizerGetOpInfo()
2289 op_info->AddSymbol.Name = name; in SymbolizerGetOpInfo()
2299 op_info->AddSymbol.Present = 1; in SymbolizerGetOpInfo()
2301 op_info->AddSymbol.Name = add; in SymbolizerGetOpInfo()
2303 op_info->AddSymbol.Value = r_value; in SymbolizerGetOpInfo()
2304 op_info->SubtractSymbol.Present = 1; in SymbolizerGetOpInfo()
[all …]

12