| /external/tensorflow/tensorflow/core/grappler/costs/ |
| op_level_cost_estimator_test.cc | 48 SetCpuDevice(&op_context.op_info);
49 op_context.op_info.set_op("MatMul");
51 DescribeMatrix(m, l, &op_context.op_info);
52 DescribeMatrix(k, n, &op_context.op_info);
59 SetCpuDevice(&op_context.op_info);
60 op_context.op_info.set_op("MatMul");
62 auto input = op_context.op_info.add_inputs();
66 input = op_context.op_info.add_inputs();
89 SetCpuDevice(&op_context.op_info);
90 op_context.op_info.set_op("BatchMatMul")
[all...] |
| op_context.h | 30 OpInfo op_info; member in struct:tensorflow::grappler::OpContext
|
| utils.cc | 80 // Annotate the op_info inputs with extra information when possible (e.g. the
85 OpInfo* op_info) {
106 if (i >= op_info->inputs_size()) {
108 << op_info->DebugString()
114 if (input_node->op() == "Const" && i < op_info->inputs_size()) {
123 OpInfo::TensorProperties* input = op_info->mutable_inputs(i);
147 (*op_info->mutable_attr())[attr_key] = attr;
158 (*op_info->mutable_attr())[new_key] = attr;
224 OpInfo op_info; local
225 op_info.set_op(node.op())
[all...] |
| utils.h | 51 string GetOpDescription(const OpInfo& op_info);
|
| op_level_cost_estimator.cc | 292 const auto& op_features = op_context.op_info;
363 const auto& op_features = op_context.op_info;
406 auto costs = PredictOpCountBasedCost(0, op_context.op_info);
[all...] |
| analytical_cost_estimator.cc | 101 for (const auto& output : op_context.op_info.outputs()) {
|
| virtual_scheduler.cc | 620 auto& op_info = op_context.op_info; local
621 op_info.set_op(node->op());
622 *op_info.mutable_attr() = node->attr();
624 *op_info.add_inputs() = input;
627 *op_info.add_outputs() = output;
629 op_info.mutable_device()->Swap(&device);
728 string node_description = GetOpDescription(op_context.op_info);
[all...] |
| virtual_scheduler_test.cc | 854 if (op_context.op_info.op() == "MatMul") {
856 } else if (op_context.op_info.op() == "RandomUniform") {
[all...] |
| /external/capstone/bindings/java/capstone/ |
| Arm.java | 135 public OpInfo(UnionOpInfo op_info) {
136 usermode = op_info.usermode;
137 vectorSize = op_info.vector_size;
138 vectorData = op_info.vector_data;
139 cpsMode = op_info.cps_mode;
140 cpsFlag = op_info.cps_flag;
141 cc = op_info.cc;
142 updateFlags = (op_info.update_flags > 0);
143 writeback = (op_info.writeback > 0);
144 memBarrier = op_info.mem_barrier
[all...] |
| Ppc.java | 102 public OpInfo(UnionOpInfo op_info) {
103 bc = op_info.bc;
104 bh = op_info.bh;
105 updateCr0 = (op_info.update_cr0 > 0);
106 op = op_info.op;
|
| Sparc.java | 86 public OpInfo(UnionOpInfo op_info) {
87 cc = op_info.cc;
88 hint = op_info.hint;
89 op = op_info.op;
|
| Arm64.java | 120 public OpInfo(UnionOpInfo op_info) {
121 cc = op_info.cc;
122 updateFlags = (op_info._update_flags > 0);
123 writeback = (op_info._writeback > 0);
124 op = op_info.op;
|
| Systemz.java | 86 public OpInfo(UnionOpInfo op_info) {
87 cc = op_info.cc;
88 op = op_info.op;
|
| Xcore.java | 79 public OpInfo(UnionOpInfo op_info) {
80 op = op_info.op;
|
| Capstone.java | 156 OpInfo op_info = null; local
162 op_info = new Arm.OpInfo((Arm.UnionOpInfo) detail.arch.arm);
167 op_info = new Arm64.OpInfo((Arm64.UnionOpInfo) detail.arch.arm64);
172 op_info = new Mips.OpInfo((Mips.UnionOpInfo) detail.arch.mips);
177 op_info = new X86.OpInfo((X86.UnionOpInfo) detail.arch.x86);
182 op_info = new Sparc.OpInfo((Sparc.UnionOpInfo) detail.arch.sparc);
187 op_info = new Systemz.OpInfo((Systemz.UnionOpInfo) detail.arch.sysz);
192 op_info = new Ppc.OpInfo((Ppc.UnionOpInfo) detail.arch.ppc);
197 op_info = new Xcore.OpInfo((Xcore.UnionOpInfo) detail.arch.xcore);
202 return op_info;
[all...] |
| /external/tensorflow/tensorflow/contrib/tensorboard/plugins/trace/ |
| trace.py | 63 op_info = trace_info.ops.add()
64 op_info.name = op.name
65 op_info.op_type = op.type
66 op_info.device = op.device
72 line_trace = op_info.traceback.add()
76 _add_data_from_tensors(op.inputs, op_info.inputs)
77 _add_data_from_tensors(op.outputs, op_info.outputs)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| static_schedule.cc | 34 op_context.op_info.set_op(node.op());
35 *op_context.op_info.mutable_attr() = node.attr();
40 op_context.op_info.add_inputs()->Swap(&input);
44 op_context.op_info.mutable_device()->Swap(&device);
|
| /external/mesa3d/src/compiler/nir/ |
| nir_builder.h | 178 const nir_op_info *op_info = &nir_op_infos[op]; local
196 unsigned num_components = op_info->output_size;
198 for (unsigned i = 0; i < op_info->num_inputs; i++) {
199 if (op_info->input_sizes[i] == 0)
209 unsigned bit_size = nir_alu_type_get_type_size(op_info->output_type);
211 for (unsigned i = 0; i < op_info->num_inputs; i++) {
213 if (nir_alu_type_get_type_size(op_info->input_types[i]) == 0) {
220 nir_alu_type_get_type_size(op_info->input_types[i]));
228 for (unsigned i = 0; i < op_info->num_inputs; i++) {
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes.cc | 648 for (const QuantizedOpInfo& op_info : op_list) {
649 if (ops_to_ignore.count(op_info.float_name) == 0) {
651 op_info.float_name);
652 op_map.insert({op_info.float_name, op_info});
701 const QuantizedOpInfo& op_info = op_map[float_node.op()];
710 if (op_info.unquantized_inputs.count(i)) {
734 if (op_info.unquantized_inputs.count(i)) {
806 for (const string& attr_to_copy : op_info.attrs_to_copy) {
[all...] |
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| virtual_cluster.cc | 84 for (const auto& output : op_context.op_info.outputs()) {
|
| /external/mesa3d/src/gallium/drivers/r600/sb/ |
| sb_pass.h | 228 struct op_info { struct in class:r600_sb::gcm
231 op_info() : top_bb(), bottom_bb() {} function in struct:r600_sb::gcm::op_info
234 typedef std::map<node*, op_info> op_info_map;
290 void bu_find_best_bb(node *n, op_info &oi);
|
| sb_gcm.cpp | 114 op_info &o = op_map[n];
559 void gcm::bu_find_best_bb(node *n, op_info &oi) {
629 op_info &oi = op_map[n];
|
| /external/llvm/tools/llvm-objdump/ |
| MachODump.cpp | 1814 struct LLVMOpInfo1 *op_info = (struct LLVMOpInfo1 *)TagBuf; local
[all...] |
| /external/iproute2/devlink/ |
| devlink.c | 3095 struct dpipe_op_info *op_info = &action->info; local
3177 struct dpipe_op_info *op_info = &match->info; local
[all...] |
| /external/mesa3d/src/gallium/state_trackers/clover/api/ |
| memory.cpp | 99 const void *op_info, cl_int *r_errcode) try {
104 auto reg = reinterpret_cast<const cl_buffer_region *>(op_info);
|
