| /external/mesa3d/src/mesa/state_tracker/ |
| st_atom_array.c | 343 for (attr = 0; attr < vpv->num_inputs; attr++) {
412 if (idx < vp->num_inputs &&
463 if (vpv->num_inputs) {
478 for (attr = 1; attr < vpv->num_inputs; attr++) {
497 for (attr = 0; attr < vpv->num_inputs;) {
523 if (vpv->num_inputs == 0) {
573 for (attr = 0; attr < vpv->num_inputs;) {
663 memset(velements, 0, sizeof(struct pipe_vertex_element) * vpv->num_inputs);
675 if (vpv->num_inputs == 0)
693 cso_set_vertex_elements(st->cso_context, vpv->num_inputs, velements)
[all...] |
| st_glsl_to_nir.cpp | 78 unsigned attr, num_inputs = 0; local
84 input_to_index[attr] = num_inputs;
85 num_inputs++;
88 num_inputs++;
95 nir->num_inputs = 0;
102 nir->num_inputs++;
332 &nir->num_inputs,
|
| st_cb_bitmap_shader.c | 77 for (i = 0; i < ctx->info.num_inputs; i++) {
86 texcoord_index = ctx->info.num_inputs;
|
| /frameworks/ml/nn/common/operations/ |
| LSTMTest.cpp | 180 uint32_t num_inputs() const { return n_input_; } function in class:android::nn::wrapper::LSTMOpModel
366 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs());
368 float* batch0_start = lstm_input + i * lstm.num_inputs();
369 float* batch0_end = batch0_start + lstm.num_inputs();
472 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs());
474 float* batch0_start = lstm_input + i * lstm.num_inputs();
475 float* batch0_end = batch0_start + lstm.num_inputs();
[all...] |
| LSHProjection.cpp | 41 const int num_inputs = NumInputsWithValues(operation, operands); local
42 NN_CHECK(num_inputs == 3 || num_inputs == 4);
|
| RNN.cpp | 45 const int num_inputs = NumInputsWithValues(operation, operands); local
46 NN_CHECK(num_inputs == 5 || num_inputs == 6);
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| unidirectional_sequence_lstm_test.cc | 208 int num_inputs() { return n_input_; } function in class:tflite::__anon39285::UnidirectionalLSTMOpModel
344 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length();
446 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length();
[all...] |
| /external/mesa3d/src/compiler/nir/ |
| nir_lower_vec_to_movs.c | 58 assert(start_idx < nir_op_infos[vec->op].num_inputs);
123 assert(start_idx < nir_op_infos[vec->op].num_inputs);
169 for (unsigned j = 0; j < nir_op_infos[src_alu->op].num_inputs; j++)
176 for (unsigned j = 0; j < nir_op_infos[src_alu->op].num_inputs; j++)
198 for (unsigned j = 0; j < nir_op_infos[src_alu->op].num_inputs; j++)
|
| /external/mesa3d/src/gallium/auxiliary/tgsi/ |
| tgsi_two_side.c | 54 uint num_inputs; member in struct:two_side_transform_context
86 ts->num_inputs = MAX2(ts->num_inputs, decl->Range.Last + 1);
114 decl.Range.First = decl.Range.Last = ts->num_inputs++;
139 decl.Range.First = decl.Range.Last = ts->num_inputs++;
|
| /external/tensorflow/tensorflow/core/graph/ |
| optimizer_cse.cc | 70 DCHECK_EQ(in->size(), n->num_inputs());
97 const int N_in = n->num_inputs();
148 if (a->num_inputs() != b->num_inputs()) return false;
149 const int N_in = a->num_inputs();
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| constant_folding_test.cc | 116 EXPECT_EQ(1, s1->num_inputs());
119 EXPECT_EQ(1, s2->num_inputs());
196 EXPECT_EQ(1, s1->num_inputs());
200 EXPECT_EQ(1, s2->num_inputs());
224 EXPECT_EQ(1, s3->num_inputs());
251 EXPECT_EQ(1, b0->num_inputs());
253 EXPECT_EQ(1, b1->num_inputs());
285 ASSERT_EQ(1, b0->num_inputs());
289 ASSERT_EQ(1, b1->num_inputs());
292 ASSERT_EQ(1, b1_ident->num_inputs());
[all...] |
| shape_refiner.cc | 78 if (index < 0 || outer_context->num_inputs() <= index) {
81 " not in [0, ", outer_context->num_inputs(), ").");
214 std::vector<Node*> input_nodes(node->num_inputs());
215 std::vector<ShapeHandle> input_shapes(node->num_inputs());
217 input_handle_shapes_and_types(node->num_inputs());
255 std::vector<const Tensor*> input_tensors(node->num_inputs(), nullptr);
628 if (current_node->num_inputs() == 0) {
730 for (int i = 0; i < src_context->num_inputs(); ++i) {
755 src_op == "Concat" ? 0 : src_context->num_inputs() - 1;
757 for (int i = 0; i < src_context->num_inputs(); ++i)
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| decode_raw_op_test.py | 92 for num_inputs in range(3):
93 result = decode.eval(feed_dict={in_bytes: [""] * num_inputs})
94 self.assertEqual((num_inputs, 0), result.shape)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| while_op.cc | 37 VLOG(2) << "Num inputs " << ctx->num_inputs();
38 args->resize(ctx->num_inputs());
41 for (int i = 0; i < ctx->num_inputs(); ++i) {
236 int num_inputs = body.input_mapping.size(); local
237 std::vector<xla::ComputationDataHandle> inputs(num_inputs);
238 for (int i = 0; i < num_inputs; ++i) {
|
| /external/tensorflow/tensorflow/c/ |
| python_api.cc | 66 if (ic_dst->num_inputs() <= dst.index) {
69 "] is greater than the number of total inputs [", ic_dst->num_inputs(),
|
| /external/tensorflow/tensorflow/cc/framework/ |
| ops.cc | 26 CHECK_LT(i, node_->num_inputs());
55 inputs.resize(node->num_inputs(), {nullptr, -1});
|
| /external/tensorflow/tensorflow/contrib/tensorrt/kernels/ |
| trt_engine_op.cc | 53 int num_binding = context->num_inputs() + context->num_outputs();
59 for (int i = 0; i < context->num_inputs(); i++) {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| in_topk_op.cc | 32 if (context->num_inputs() == 2) {
41 if (context->num_inputs() == 3) {
|
| save_op.cc | 57 for (int i = 0; i < ctx->num_inputs(); ++i) {
80 for (int i = 0; i < ctx->num_inputs(); ++i) {
|
| bcast_ops.cc | 32 ctx, ctx->num_inputs() == 2,
35 for (int i = 0; i < ctx->num_inputs(); ++i) {
81 ctx, ctx->num_inputs() == 2,
84 for (int i = 0; i < ctx->num_inputs(); ++i) {
|
| save_restore_v2_ops.cc | 74 OP_REQUIRES(context, context->num_inputs() == num_tensors + kFixedInputs,
77 context->num_inputs() - kFixedInputs, " tensors."));
78 OP_REQUIRES(context, context->num_inputs() == num_tensors + kFixedInputs,
83 num_tensors, " names, but received ", context->num_inputs(),
|
| function_ops.cc | 150 OP_REQUIRES(ctx, ctx->num_inputs() == ctx->num_outputs(),
151 errors::Internal("#inputs != #outputs : ", ctx->num_inputs(),
153 for (int i = 0; i < ctx->num_inputs(); ++i) {
163 for (int i = 0; i < ctx->num_inputs(); ++i) {
251 args.reserve(ctx->num_inputs());
252 for (int i = 0; i < ctx->num_inputs(); ++i) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| string_ops.cc | 72 for (int i = 0; i < c->num_inputs(); ++i) {
85 for (int i = 0; i < c->num_inputs(); ++i) {
|
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 719 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs()))
722 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs()))
731 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs()))
734 ctext->input(GetTensorMetaDataIndex(n, ctext->num_inputs()))
743 return ctext->input(GetTensorDataIndex(n, ctext->num_inputs()));
789 CHECK_LT(input_idx, context->num_inputs());
978 int num_inputs = context->num_inputs(); local
1001 int num_inputs = context->num_inputs(); local
1019 int num_inputs = context->num_inputs(); local
1039 int num_inputs = context->num_inputs(); local
1058 int num_inputs = context->num_inputs(); local
1077 int num_inputs = context->num_inputs(); local
1097 int num_inputs = context->num_inputs(); local
[all...] |
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| lp_bld_interp.h | 117 unsigned num_inputs,
|
