| /external/tensorflow/tensorflow/python/ops/ |
| control_flow_grad.py | 128 num_inputs = len(op.inputs)
129 cond = [math_ops.equal(op.outputs[1], i) for i in xrange(num_inputs)]
132 for i in xrange(num_inputs)]
|
| math_ops_test.py | 327 for num_inputs in range(1, 10):
328 x = [np.random.random((1, 2, 3, 4, 5)) - 0.5 for _ in range(num_inputs)]
332 self.assertAllClose(x[0] * num_inputs,
333 math_ops.add_n([tf_x[0]] * num_inputs).eval())
337 for num_inputs in range(1, 10):
340 for _ in range(num_inputs)
345 self.assertAllEqual(x[0] * num_inputs,
346 math_ops.add_n([tf_x[0]] * num_inputs).eval())
350 for num_inputs in range(1, 10):
354 for i in range(0, num_inputs)
[all...] |
| /frameworks/ml/nn/common/operations/ |
| SVDF.cpp | 60 const int num_inputs = NumInputsWithValues(operation, operands); local
62 NN_CHECK(num_inputs == 6 || num_inputs == 7);
|
| /external/tensorflow/tensorflow/core/ops/ |
| io_ops.cc | 31 for (int i = 0; i < c->num_inputs(); ++i) {
43 for (int i = 0; i < c->num_inputs(); ++i) {
79 c->WithValue(c->Dim(s, 0), c->num_inputs() - 3, &unused_dim));
161 c->WithValue(c->Dim(s, 0), c->num_inputs() - 2, &unused_dim));
185 c->WithValue(c->Dim(s, 0), c->num_inputs() - 3, &unused_dim));
|
| /external/tensorflow/tensorflow/contrib/lite/toco/ |
| import_tensorflow.cc | 615 const int num_inputs = GetInputsCount(node, tf_import_flags); local
616 QCHECK(num_inputs == 3 || num_inputs == 4)
705 const int num_inputs = GetInputsCount(node, tf_import_flags); local
707 for (int i = 0; i < num_inputs; ++i) {
1025 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1043 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1056 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1069 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1082 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1095 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1108 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1175 const int num_inputs = GetInputsCount(node, tf_import_flags); local
1629 const int num_inputs = GetInputsCount(node, tf_import_flags); local
[all...] |
| /external/mesa3d/src/mesa/state_tracker/ |
| st_program.c | 243 stvp->num_inputs = 0;
254 input_to_index[attr] = stvp->num_inputs;
255 stvp->index_to_input[stvp->num_inputs] = attr;
256 stvp->num_inputs++;
260 stvp->index_to_input[stvp->num_inputs] = ST_DOUBLE_ATTRIB_PLACEHOLDER;
261 stvp->num_inputs++;
266 input_to_index[VERT_ATTRIB_EDGEFLAG] = stvp->num_inputs;
267 stvp->index_to_input[stvp->num_inputs] = VERT_ATTRIB_EDGEFLAG;
423 stvp->num_inputs,
448 stvp->num_inputs,
1204 uint num_inputs = 0; local
[all...] |
| st_debug.c | 94 for (i = 0; i < st->vp->state.num_inputs; i++) {
|
| st_cb_drawpixels_shader.c | 107 for (i = 0; i < ctx->info.num_inputs; i++) {
116 texcoord_index = ctx->info.num_inputs;
|
| st_program.h | 195 GLuint num_inputs; member in struct:st_vp_variant
215 GLuint num_inputs; member in struct:st_vertex_program
|
| /external/mesa3d/src/compiler/nir/ |
| nir_search.c | 268 assert(nir_op_infos[instr->op].num_inputs > 0);
289 for (unsigned i = 0; i < nir_op_infos[instr->op].num_inputs; i++) {
301 assert(nir_op_infos[instr->op].num_inputs == 2);
342 tree->num_srcs = info.num_inputs;
344 for (unsigned i = 0; i < info.num_inputs; i++) {
464 for (unsigned i = 0; i < nir_op_infos[expr->opcode].num_inputs; i++) {
|
| nir_lower_alu_to_scalar.c | 54 if (nir_op_infos[chan_op].num_inputs > 1) {
55 assert(nir_op_infos[chan_op].num_inputs == 2);
79 unsigned num_src = nir_op_infos[instr->op].num_inputs;
|
| nir_opt_undef.c | 89 for (unsigned i = 0; i < nir_op_infos[alu->op].num_inputs; i++) {
|
| /external/mesa3d/src/gallium/auxiliary/tgsi/ |
| tgsi_scan.h | 47 ubyte num_inputs; member in struct:tgsi_shader_info
|
| /external/mesa3d/src/gallium/drivers/ilo/shader/ |
| ilo_shader_internal.h | 53 int num_inputs; member in struct:ilo_shader_variant::__anon29576::__anon29578
|
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| lp_setup_context.h | 213 unsigned num_inputs,
|
| /external/mesa3d/src/gallium/drivers/vc4/ |
| vc4_opt_dead_code.c | 139 c->num_inputs--;
|
| vc4_context.h | 143 uint32_t num_inputs; member in struct:vc4_fs_inputs
175 uint8_t num_inputs; member in struct:vc4_compiled_shader
|
| /external/tensorflow/tensorflow/contrib/ffmpeg/ |
| decode_video_op.cc | 40 context, context->num_inputs() == 1,
|
| decode_audio_op.cc | 111 context, context->num_inputs() == 4,
240 context, context->num_inputs() == 1,
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph.cc | 145 int32 Node::num_inputs() const { return props_->input_types.size(); } function in class:tensorflow::Node
196 if (idx < 0 || idx >= num_inputs()) {
198 name(), " only has ", num_inputs(),
224 input_edges->resize(num_inputs(), nullptr);
228 if (edge->dst_input() < 0 || edge->dst_input() >= num_inputs()) {
238 for (int i = 0; i < num_inputs(); ++i) {
555 inputs.resize(node->num_inputs(), nullptr);
627 if (idx >= node->num_inputs()) {
630 "', num of inputs: ", node->num_inputs(),
|
| /external/tensorflow/tensorflow/core/kernels/ |
| fake_quant_ops.cc | 190 CHECK_EQ(3, context->num_inputs());
229 CHECK_EQ(4, context->num_inputs());
319 CHECK_EQ(3, context->num_inputs());
366 CHECK_EQ(4, context->num_inputs());
|
| remote_fused_graph_execute_op.cc | 76 const int input_count = ctx->num_inputs();
|
| /external/mesa3d/src/gallium/auxiliary/draw/ |
| draw_vs_exec.c | 121 for (slot = 0; slot < shader->info.num_inputs; slot++) {
148 for (slot = 0; slot < shader->info.num_inputs; slot++) {
|
| /external/tensorflow/tensorflow/cc/framework/ |
| while_gradients_test.cc | 34 void Init(int num_inputs, DataType dtype = DT_INT32) {
35 for (int i = 0; i < num_inputs; ++i) {
|
| /external/tensorflow/tensorflow/cc/ops/ |
| while_loop_test.cc | 32 void Init(int num_inputs, DataType dtype = DT_INT32) {
33 for (int i = 0; i < num_inputs; ++i) {
|
