Searched refs:num_inputs (Results 1 - 25 of 247) sorted by relevance

12345678910

/external/mesa3d/prebuilt-intermediates/nir/
H A Dnir_opcodes.c7 .num_inputs = 1,
21 .num_inputs = 1,
35 .num_inputs = 2,
49 .num_inputs = 2,
63 .num_inputs = 2,
77 .num_inputs = 2,
91 .num_inputs = 2,
105 .num_inputs = 2,
119 .num_inputs = 2,
133 .num_inputs
[all...]
/external/mesa3d/src/gallium/drivers/svga/
H A Dsvga_link.h11 unsigned num_inputs; member in struct:shader_linkage
H A Dsvga_link.c58 assert(inshader_info->num_inputs <=
64 for (i = 0; i < inshader_info->num_inputs; i++) {
90 linkage->num_inputs = inshader_info->num_inputs;
93 for (i = 0; i < inshader_info->num_inputs; i++) {
105 for (i = 0; i < linkage->num_inputs; i++) {
/external/tensorflow/tensorflow/python/ops/
H A Dconcat_benchmark.py35 def build_graph(device, input_shape, variable, num_inputs, axis, grad):
42 num_inputs: the number of inputs to concat
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,
89 num_inputs: the number of inputs to concat
99 outputs = build_graph(device, input_shape, variable, num_inputs, axis,
113 num_inputs, axis, grad, duration / num_iters,
114 num_inputs * input_shap
[all...]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Daggregate_ops.cc29 OP_REQUIRES(ctx, ctx->num_inputs() >= 1,
33 for (int i = 1; i < ctx->num_inputs(); ++i) {
H A Dfunction_ops.cc32 OP_REQUIRES(ctx, ctx->num_inputs() == ctx->num_outputs(),
33 errors::Internal("#inputs != #outputs : ", ctx->num_inputs(),
35 for (int i = 0; i < ctx->num_inputs(); ++i) {
45 for (int i = 0; i < ctx->num_inputs(); ++i) {
80 args.reserve(ctx->num_inputs());
81 for (int i = 0; i < ctx->num_inputs(); ++i) {
H A Didentity_op.cc27 for (int i = 0; i < ctx->num_inputs(); ++i) {
H A Dbcast_ops.cc39 ctx, ctx->num_inputs() == 2,
42 for (int i = 0; i < ctx->num_inputs(); ++i) {
87 ctx, ctx->num_inputs() == 2,
91 for (int i = 0; i < ctx->num_inputs(); ++i) {
/external/tensorflow/tensorflow/core/ops/
H A Dfunctional_ops_test.cc26 int num_inputs = 4; local
31 for (int i = 0; i < num_inputs; ++i) {
H A Dfunctional_ops.cc31 if (c->num_inputs() < c->num_outputs()) {
/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_cpu.h35 size_t num_inputs = inputs.size(); local
38 sizes.reserve(num_inputs);
58 inp.reserve(num_inputs);
64 for (int64 j = 0; j < num_inputs; ++j) {
75 auto work = [&row_size, &sizes, &inputs, &output, &copier, &num_inputs](
84 for (size_t j = 0; j < num_inputs; ++j) {
110 inp.reserve(num_inputs);
116 for (int64 j = 0; j < num_inputs; ++j) {
137 size_t num_inputs = inputs.size(); local
140 sizes.reserve(num_inputs);
[all...]
H A Dlogging_ops.cc40 for (int i = 1; i < ctx->num_inputs(); ++i) {
43 if (i < ctx->num_inputs() - 1) strings::StrAppend(&msg, " ");
76 for (int i = 1; i < ctx->num_inputs(); ++i) {
H A Dsave_op.cc57 for (int i = 0; i < ctx->num_inputs(); ++i) {
80 for (int i = 0; i < ctx->num_inputs(); ++i) {
/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dremove_nodes.cc56 for (int num_inputs = 1; num_inputs <= max_inputs; ++num_inputs) {
59 pattern.inputs.resize(num_inputs);
60 for (int i = 0; i < num_inputs; ++i) {
/external/tensorflow/tensorflow/compiler/tests/
H A Dlstm.py40 def LSTMCellWeightsShape(num_inputs, num_nodes):
44 return [num_inputs + num_nodes, 4 * num_nodes]
54 x: Input with shape [batch_size, num_inputs].
96 x_seq: List of inputs, each with shape [batch_size, num_inputs].
124 def RandomInputs(batch_size, seq_length, num_inputs):
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):
143 num_inputs: Dimension of inputs that are fed into each LSTM cell.
152 LSTMCellWeightsShape(num_inputs, num_nodes), name='weights')
155 x_seq, pad_seq = RandomInputs(batch_size, seq_length, num_inputs)
[all...]
/external/tensorflow/tensorflow/contrib/framework/python/ops/
H A Daccumulate_n_v2_eager_test.py60 num_inputs = 3
64 for i in range(0, num_inputs)
72 self.assertAllEqual(np.repeat(1.0, num_inputs), # d/dx (x + y + ...) = 1
/external/tensorflow/tensorflow/contrib/tpu/ops/
H A Dreplication_ops.cc38 ShapeHandle cur = c->input(c->num_inputs() - 1);
39 for (int i = c->num_inputs() - 2; i >= 0; --i) {
H A Dtpu_configuration_ops.cc86 for (int i = 0; i < c->num_inputs(); ++i) {
112 for (int i = 0; i < c->num_inputs(); ++i) {
/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_compilation_cache.cc101 signature->arg_types.reserve(ctx->num_inputs() - num_constant_args);
111 while (input_num < ctx->num_inputs() - variable_args.size()) {
119 TF_RET_CHECK(input_num < ctx->num_inputs());
139 args->resize(ctx->num_inputs());
144 TF_RET_CHECK(num_constant_args <= ctx->num_inputs());
159 ctx->num_inputs() - num_variable_args - num_constant_args;
177 TF_RET_CHECK(input_num + num_variable_args == ctx->num_inputs());
244 VLOG(2) << "num_inputs=" << ctx->num_inputs()
247 for (int i = 0; i < ctx->num_inputs();
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dlstm_test.cc204 int num_inputs() { return n_input_; } function in class:tflite::__anon25943::LSTMOpModel
337 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs());
339 float* batch0_start = lstm_input + i * lstm.num_inputs();
340 float* batch0_end = batch0_start + lstm.num_inputs();
442 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs());
444 float* batch0_start = lstm_input + i * lstm.num_inputs();
445 float* batch0_end = batch0_start + lstm.num_inputs();
1056 sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs());
1058 float* batch0_start = lstm_input[0] + i * lstm.num_inputs();
1059 float* batch0_end = batch0_start + lstm.num_inputs();
[all...]
/external/mesa3d/src/mesa/state_tracker/
H A Dst_draw_feedback.c172 if (vp->num_inputs) {
175 for (attr = 1; attr < vp->num_inputs; attr++) {
184 for (attr = 0; attr < vp->num_inputs; attr++) {
238 draw_set_vertex_buffers(draw, 0, vp->num_inputs, vbuffers);
239 draw_set_vertex_elements(draw, vp->num_inputs, velements);
291 for (attr = 0; attr < vp->num_inputs; attr++) {
297 draw_set_vertex_buffers(draw, 0, vp->num_inputs, NULL);
/external/mesa3d/src/compiler/nir/
H A Dnir_move_vec_src_uses_to_dest.c94 for (unsigned i = 0; i < nir_op_infos[vec->op].num_inputs; i++) {
113 for (unsigned j = i; j < nir_op_infos[vec->op].num_inputs; j++) {
143 assert(src_idx < nir_op_infos[use_alu->op].num_inputs);
/external/mesa3d/src/gallium/drivers/llvmpipe/
H A Dlp_state_setup.h19 unsigned num_inputs:8; member in struct:lp_setup_variant_key
/external/tensorflow/tensorflow/core/framework/
H A Dshape_inference_testutil_test.cc52 const int num_inputs = 1 + std::count(ins.begin(), ins.end(), ';'); local
54 src_list.reserve(num_inputs);
55 for (int i = 0; i < num_inputs; ++i) src_list.emplace_back("a", 0, DT_FLOAT);
59 .Attr("N", num_inputs)
/external/mesa3d/src/gallium/drivers/ilo/shader/
H A Dtoy_tgsi.h76 int num_inputs; member in struct:toy_tgsi
108 for (slot = 0; slot < tgsi->num_inputs; slot++) {

Completed in 605 milliseconds

12345678910