| /external/mesa3d/prebuilt-intermediates/nir/ |
| nir_opcodes.c | 7 .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 = 2
[all...] |
| /frameworks/ml/nn/common/operations/ |
| Concatenation.cpp | 28 int num_inputs = inputShapes.size(); local
29 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs);
30 std::vector<tflite::Dims<4> > inputDims(num_inputs);
31 for (int i=0; i<num_inputs; i++) {
38 inputDataPtrs.data(), inputDimsPtr.data(), num_inputs, local
47 int num_inputs = inputShapes.size(); local
48 std::vector<tflite::Dims<4>*> inputDimsPtr(num_inputs);
49 std::vector<tflite::Dims<4> > inputDims(num_inputs);
50 for (int i=0; i<num_inputs; i++) {
57 inputDataPtrs.data(), inputDimsPtr.data(), num_inputs, local
[all...] |
| /external/mesa3d/src/gallium/drivers/svga/ |
| svga_link.h | 11 unsigned num_inputs; member in struct:shader_linkage
|
| svga_link.c | 58 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/ |
| concat_benchmark.py | 35 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_shape[0] * input_shape[1] * 4 * 2
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| aggregate_ops.cc | 29 OP_REQUIRES(ctx, ctx->num_inputs() >= 1,
33 for (int i = 1; i < ctx->num_inputs(); ++i) {
|
| function_ops.cc | 32 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) {
|
| identity_op.cc | 27 for (int i = 0; i < ctx->num_inputs(); ++i) {
|
| bcast_ops.cc | 39 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/ |
| functional_ops_test.cc | 26 int num_inputs = 4; local
31 for (int i = 0; i < num_inputs; ++i) {
|
| functional_ops.cc | 31 if (c->num_inputs() < c->num_outputs()) {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| concat_lib_cpu.h | 35 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...] |
| logging_ops.cc | 40 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) {
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_nodes.cc | 56 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/ |
| lstm.py | 40 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/tpu/ops/ |
| replication_ops.cc | 38 ShapeHandle cur = c->input(c->num_inputs() - 1);
39 for (int i = c->num_inputs() - 2; i >= 0; --i) {
|
| tpu_configuration_ops.cc | 86 for (int i = 0; i < c->num_inputs(); ++i) {
112 for (int i = 0; i < c->num_inputs(); ++i) {
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_compilation_cache.cc | 101 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(); i++)
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| lstm_test.cc | 204 int num_inputs() { return n_input_; } function in class:tflite::__anon39264::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();
[all...] |
| /external/mesa3d/src/mesa/state_tracker/ |
| st_draw_feedback.c | 172 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/ |
| nir_move_vec_src_uses_to_dest.c | 94 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/tensorflow/tensorflow/contrib/framework/python/ops/ |
| accumulate_n_v2_eager_test.py | 60 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/mesa3d/src/gallium/drivers/llvmpipe/ |
| lp_state_setup.h | 19 unsigned num_inputs:8; member in struct:lp_setup_variant_key
|
| /external/tensorflow/tensorflow/core/framework/ |
| shape_inference_testutil_test.cc | 52 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/ |
| toy_tgsi.h | 76 int num_inputs; member in struct:toy_tgsi
108 for (slot = 0; slot < tgsi->num_inputs; slot++) {
|
