| /external/mesa3d/prebuilt-intermediates/nir/ |
| D | nir_opcodes.c | 173 .num_inputs = 1,
187 .num_inputs = 1,
201 .num_inputs = 2,
215 .num_inputs = 2,
229 .num_inputs = 2,
243 .num_inputs = 2,
257 .num_inputs = 2,
271 .num_inputs = 2,
285 .num_inputs = 2,
299 .num_inputs = 2,
[all …]
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | ml_sse3.c | 63 const int num_inputs) { in nn_propagate_4to4() argument
68 const __m128 weight0 = _mm_loadu_ps(&weights[2 * i * num_inputs]); in nn_propagate_4to4()
70 const __m128 weight1 = _mm_loadu_ps(&weights[(2 * i + 1) * num_inputs]); in nn_propagate_4to4()
85 __m128 *const out_l, const int num_inputs) { in nn_propagate_4to8() argument
90 const __m128 weight0 = _mm_loadu_ps(&weights[2 * i * num_inputs]); in nn_propagate_4to8()
91 const __m128 weight1 = _mm_loadu_ps(&weights[(2 * i + 1) * num_inputs]); in nn_propagate_4to8()
112 const int num_inputs) { in nn_propagate_8to4() argument
119 const __m128 weight_h = _mm_loadu_ps(&weights[i * num_inputs + 4]); in nn_propagate_8to4()
120 const __m128 weight_l = _mm_loadu_ps(&weights[i * num_inputs]); in nn_propagate_8to4()
158 int num_inputs = nn_config->num_inputs; in av1_nn_predict_sse3() local
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | concat_benchmark.py | 35 def build_graph(device, input_shape, variable, num_inputs, axis, grad): argument
51 inputs = [array_ops.zeros(input_shape) for _ in range(num_inputs)]
58 ]) for _ in range(num_inputs)
65 ]) for _ in range(num_inputs)
81 def _run_graph(self, device, input_shape, variable, num_inputs, axis, grad, argument
99 outputs = build_graph(device, input_shape, variable, num_inputs, axis,
113 num_inputs, axis, grad, duration / num_iters,
114 num_inputs * input_shape[0] * input_shape[1] * 4 * 2 *
123 num_inputs=num_inputs,
137 num_inputs = 20
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | concat_lib_cpu.h | 38 size_t num_inputs = inputs.size(); in ConcatCPUImpl() local
41 sizes.reserve(num_inputs); in ConcatCPUImpl()
61 inp.reserve(num_inputs); in ConcatCPUImpl()
67 for (int64 j = 0; j < num_inputs; ++j) { in ConcatCPUImpl()
78 auto work = [&row_size, &sizes, &inputs, &output, &copier, &num_inputs]( in ConcatCPUImpl()
87 for (size_t j = 0; j < num_inputs; ++j) { in ConcatCPUImpl()
113 inp.reserve(num_inputs); in ConcatCPUImpl()
119 for (int64 j = 0; j < num_inputs; ++j) { in ConcatCPUImpl()
140 size_t num_inputs = inputs.size(); in ConcatSYCLImpl() local
143 sizes.reserve(num_inputs); in ConcatSYCLImpl()
[all …]
|
| D | searchsorted_op.cc | 38 int batch_size, int num_inputs, int num_values, in Compute()
42 const T* sorted_inputs_ptr = sorted_inputs.data() + b * num_inputs; in Compute()
46 std::upper_bound(sorted_inputs_ptr, sorted_inputs_ptr + num_inputs, in Compute()
61 int batch_size, int num_inputs, int num_values, in Compute()
65 const T* sorted_inputs_ptr = sorted_inputs.data() + b * num_inputs; in Compute()
69 std::lower_bound(sorted_inputs_ptr, sorted_inputs_ptr + num_inputs, in Compute()
|
| D | string_format_op.cc | 35 OP_REQUIRES(ctx, ctx->num_inputs() == num_placeholders, in StringFormatOp()
38 num_placeholders, " vs. ", ctx->num_inputs()))); in StringFormatOp()
48 for (int i = 0; i < ctx->num_inputs(); ++i) { in Compute()
|
| D | searchsorted_op_gpu.cu.cc | 65 int batch_size, int num_inputs, int num_values, in Compute()
74 num_inputs, num_values, values.data(), output->data())); in Compute()
85 int batch_size, int num_inputs, int num_values, in Compute()
94 num_inputs, num_values, values.data(), output->data())); in Compute()
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_atom_array.c | 320 unsigned num_inputs) in is_interleaved_arrays() argument
328 for (attr = 0; attr < num_inputs; attr++) { in is_interleaved_arrays()
401 if (idx < vp->num_inputs && in init_velement_lowered()
458 unsigned num_inputs) in setup_interleaved_attribs() argument
471 if (num_inputs) { in setup_interleaved_attribs()
486 for (attr = 1; attr < num_inputs; attr++) { in setup_interleaved_attribs()
505 for (attr = 0; attr < num_inputs;) { in setup_interleaved_attribs()
531 if (num_inputs == 0) { in setup_interleaved_attribs()
563 set_vertex_attribs(st, &vbuffer, num_inputs ? 1 : 0, in setup_interleaved_attribs()
564 velements, num_inputs); in setup_interleaved_attribs()
[all …]
|
| D | st_draw_feedback.c | 171 if (vp->num_inputs) { in st_feedback_draw_vbo()
174 for (attr = 1; attr < vp->num_inputs; attr++) { in st_feedback_draw_vbo()
183 for (attr = 0; attr < vp->num_inputs; attr++) { in st_feedback_draw_vbo()
237 draw_set_vertex_buffers(draw, 0, vp->num_inputs, vbuffers); in st_feedback_draw_vbo()
238 draw_set_vertex_elements(draw, vp->num_inputs, velements); in st_feedback_draw_vbo()
287 for (attr = 0; attr < vp->num_inputs; attr++) { in st_feedback_draw_vbo()
293 draw_set_vertex_buffers(draw, 0, vp->num_inputs, NULL); in st_feedback_draw_vbo()
|
| /external/mesa3d/src/gallium/drivers/svga/ |
| D | svga_link.c | 58 assert(inshader_info->num_inputs <= in svga_link_shaders()
64 for (i = 0; i < inshader_info->num_inputs; i++) { in svga_link_shaders()
90 linkage->num_inputs = inshader_info->num_inputs; in svga_link_shaders()
93 for (i = 0; i < inshader_info->num_inputs; i++) { in svga_link_shaders()
105 for (i = 0; i < linkage->num_inputs; i++) { in svga_link_shaders()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lstm.py | 40 def LSTMCellWeightsShape(num_inputs, num_nodes): argument
44 return [num_inputs + num_nodes, 4 * num_nodes]
124 def RandomInputs(batch_size, seq_length, num_inputs): argument
130 x_seq.append(RandomVar([batch_size, num_inputs], name='x_seq_%d' % seq))
137 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
152 LSTMCellWeightsShape(num_inputs, num_nodes), name='weights')
155 x_seq, pad_seq = RandomInputs(batch_size, seq_length, num_inputs)
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 94 kernel_state.num_inputs = n->num_inputs(); in Initialize()
103 previous_kernel_state.num_inputs; in Initialize()
141 last_kernel_state.input_start_index + last_kernel_state.num_inputs; in Initialize()
244 const size_t num_inputs = kernel_state.num_inputs; in RunAsync() local
248 node_inputs.resize(num_inputs); in RunAsync()
250 input_alloc_attrs.resize(num_inputs); in RunAsync()
251 for (size_t j = 0; j < num_inputs; ++j) { in RunAsync()
258 input_device_contexts.resize(num_inputs); in RunAsync()
289 for (size_t j = 0; j < num_inputs; ++j) { in RunAsync()
330 size_t num_inputs; member
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | function_ops.cc | 33 OP_REQUIRES(ctx, ctx->num_inputs() == ctx->num_outputs(), in PassOn()
34 errors::Internal("#inputs != #outputs : ", ctx->num_inputs(), in PassOn()
36 for (int i = 0; i < ctx->num_inputs(); ++i) { in PassOn()
46 for (int i = 0; i < ctx->num_inputs(); ++i) { in Compile()
81 args.reserve(ctx->num_inputs()); in ComputeAsync()
82 for (int i = 0; i < ctx->num_inputs(); ++i) { in ComputeAsync()
|
| D | while_op.cc | 46 VLOG(2) << "Num inputs " << ctx->num_inputs(); in MakeXlaCompilerArgumentsFromInputs()
47 args->resize(ctx->num_inputs()); in MakeXlaCompilerArgumentsFromInputs()
51 for (int i = 0; i < ctx->num_inputs(); ++i) { in MakeXlaCompilerArgumentsFromInputs()
127 std::vector<bool> loop_invariants(ctx->num_inputs()); in ConvertLoopInvariantsToConst()
129 for (int i = 0; i < ctx->num_inputs(); i++) { in ConvertLoopInvariantsToConst()
168 body.xla_output_shape.tuple_shapes(ctx->num_inputs()); in VerifyBodyInputAndOutputShapeMatch()
235 for (int i = 0; i < ctx->num_inputs(); i++) { in BuildWhile()
247 while_result, ctx->num_inputs() - num_compile_time_const_args)); in BuildWhile()
295 std::vector<bool> compile_time_const_arg_indices(ctx->num_inputs()); in Compile()
465 int num_inputs = body.input_mapping.size(); in Compile() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_dataset_op.cc | 54 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
67 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
95 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
189 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in Initialize() local
190 i < num_inputs; ++i) { in Initialize()
300 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in StartThreads() local
301 i < num_inputs; ++i) { in StartThreads()
331 for (size_t i = 1, num_inputs = histograms_.size(); i < num_inputs; in SelectFastestInputIndex() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n.cc | 30 int num_inputs = NumInputs(node); in Prepare() local
31 TF_LITE_ENSURE(context, num_inputs >= 2); in Prepare()
39 for (int i = kInputTensor1 + 1; i < num_inputs; ++i) { in Prepare()
57 int num_inputs = NumInputs(node); in EvalAddN() local
59 reference_ops::AddN<T>(GetTensorShape(input1), num_inputs, all_inputs.data(), in EvalAddN()
|
| D | while.cc | 110 int num_inputs = node->inputs->size; in Prepare() local
112 TF_LITE_ENSURE_EQ(context, node->outputs->size, num_inputs); in Prepare()
124 TF_LITE_ENSURE_EQ(context, cond_subgraph->inputs().size(), num_inputs); in Prepare()
128 TF_LITE_ENSURE_EQ(context, body_subgraph->inputs().size(), num_inputs); in Prepare()
129 TF_LITE_ENSURE_EQ(context, body_subgraph->outputs().size(), num_inputs); in Prepare()
158 for (int i = 0; i < num_inputs; ++i) { in Prepare()
178 for (int i = 0; i < num_inputs; ++i) { in Prepare()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | remove_nodes.cc | 55 for (int num_inputs = 1; num_inputs <= max_inputs; ++num_inputs) { in RemoveNodes() local
58 pattern.inputs.resize(num_inputs); in RemoveNodes()
59 for (int i = 0; i < num_inputs; ++i) { in RemoveNodes()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | mkl_array_ops.cc | 50 const int n = (c->num_inputs() - 1) / 3; in __anond4979b3e0102()
53 for (int i = n + 1; i < c->num_inputs(); ++i) { in __anond4979b3e0102()
80 const int n = (c->num_inputs() / 2 - 1) / 3; in __anond4979b3e0202()
83 for (int i = n + 1; i < c->num_inputs() / 2; ++i) { in __anond4979b3e0202()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_concat_ops.py | 237 num_inputs = len(rt_inputs)
256 row_indices = math_ops.range(rt_nrows * num_inputs)
257 row_index_matrix = array_ops.reshape(row_indices, [num_inputs, -1])
264 stack_splits = math_ops.range(0, rt_nrows * num_inputs + 1, num_inputs)
271 concat_splits = permuted_rt.row_splits[::num_inputs]
|
| /external/v8/tools/ |
| D | bigint-tester.py | 293 def RunOne(op, num_inputs, binary): argument
294 return OPS[op]().RunTest(num_inputs, binary)
300 yield (op, args.num_inputs, args.binary)
336 if args.num_inputs < 0: args.num_inputs = kNumInputsGenerate
338 OPS[op]().PrintTest(args.num_inputs)
340 if args.num_inputs < 0: args.num_inputs = kNumInputsStress
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 129 EXPECT_EQ(1, s1->num_inputs()); in TEST_F()
132 EXPECT_EQ(1, s2->num_inputs()); in TEST_F()
209 EXPECT_EQ(1, s1->num_inputs()); in TEST_F()
213 EXPECT_EQ(1, s2->num_inputs()); in TEST_F()
237 EXPECT_EQ(1, s3->num_inputs()); in TEST_F()
264 EXPECT_EQ(1, b0->num_inputs()); in TEST_F()
266 EXPECT_EQ(1, b1->num_inputs()); in TEST_F()
298 ASSERT_EQ(1, b0->num_inputs()); in TEST_F()
302 ASSERT_EQ(1, b1->num_inputs()); in TEST_F()
305 ASSERT_EQ(1, b1_ident->num_inputs()); in TEST_F()
[all …]
|
| D | shape_refiner.cc | 79 if (index < 0 || outer_context->num_inputs() <= index) { in InferShapesForFunctionSubNode()
82 " not in [0, ", outer_context->num_inputs(), ")."); in InferShapesForFunctionSubNode()
215 std::vector<const Node*> input_nodes(node->num_inputs()); in AddNode()
216 std::vector<ShapeHandle> input_shapes(node->num_inputs()); in AddNode()
218 input_handle_shapes_and_types(node->num_inputs()); in AddNode()
255 std::vector<const Tensor*> input_tensors(node->num_inputs(), nullptr); in AddNode()
500 for (int i = 0; i < src_context->num_inputs(); ++i) { in ConstantPartialShape()
517 src_op == "Concat" ? 0 : src_context->num_inputs() - 1; in ConstantPartialShape()
519 for (int i = 0; i < src_context->num_inputs(); ++i) { in ConstantPartialShape()
625 std::vector<const Tensor*> input_tensors(node->num_inputs(), nullptr); in RunShapeFn()
[all …]
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | optimizer_cse.cc | 71 DCHECK_EQ(in->size(), n->num_inputs()); in FillInputs()
98 const int N_in = n->num_inputs(); in NodeHash()
149 if (a->num_inputs() != b->num_inputs()) return false; in Equivalent()
150 const int N_in = a->num_inputs(); in Equivalent()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 266 inputs.reserve(ctx->num_inputs()); in ExecuteNativeSegment()
267 for (int i = 0; i < ctx->num_inputs(); i++) { in ExecuteNativeSegment()
304 int num_inputs = ctx->num_inputs(); in ExecuteCalibration() local
307 for (int i = 0; i < num_inputs; i++) { in ExecuteCalibration()
373 input_shapes.reserve(ctx->num_inputs()); in ComputeAsync()
374 for (int i = 0; i < ctx->num_inputs(); ++i) { in ComputeAsync()
402 const int num_binding = ctx->num_inputs() + ctx->num_outputs(); in ExecuteTrtEngine()
404 for (int i = 0; i < ctx->num_inputs(); i++) { in ExecuteTrtEngine()
661 const int num_inputs = ctx->num_inputs(); in AllocateCalibrationResources() local
663 cres->device_tensors_.resize(num_inputs); in AllocateCalibrationResources()
[all …]
|