Home
last modified time | relevance | path

Searched refs:num_inputs (Results 1 – 25 of 413) sorted by relevance

12345678910>>...17

/external/mesa3d/prebuilt-intermediates/nir/
Dnir_opcodes.c297 .num_inputs = 2,
312 .num_inputs = 2,
327 .num_inputs = 2,
342 .num_inputs = 2,
357 .num_inputs = 2,
372 .num_inputs = 2,
387 .num_inputs = 2,
402 .num_inputs = 2,
417 .num_inputs = 2,
432 .num_inputs = 2,
[all …]
/external/libaom/libaom/av1/encoder/x86/
Dml_sse3.c63 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/
Dconcat_benchmark.py35 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/
Dmkl_aggregate_ops.cc50 const int num_inputs = ctx->num_inputs() / 2; in CheckInputShape() local
53 for (size_t i = 1; i < num_inputs; ++i) { in CheckInputShape()
72 const int num_inputs = ctx->num_inputs() / 2; in FindMKLInputIndex() local
75 for (size_t i = 0; i < num_inputs; ++i) { in FindMKLInputIndex()
87 const int num_inputs = ctx->num_inputs() / 2; in ComputeScalar() local
94 for (int src_idx = 0; src_idx < num_inputs; ++src_idx) { in ComputeScalar()
113 const int num_inputs = ctx->num_inputs() / 2; in Compute() local
119 if (num_inputs == 1) { in Compute()
156 std::vector<float> coeff(num_inputs, 1.0); in Compute()
158 std::vector<MklDnnData<T>> srcs(num_inputs, MklDnnData<T>(&cpu_engine)); in Compute()
[all …]
Dconcat_lib_cpu.h38 size_t num_inputs = inputs.size(); in ConcatCPUImpl() local
41 sizes.reserve(num_inputs); in ConcatCPUImpl()
62 inp.reserve(num_inputs); in ConcatCPUImpl()
68 for (int64 j = 0; j < num_inputs; ++j) { in ConcatCPUImpl()
79 auto work = [&row_size, &sizes, &inputs, &output, &copier, &num_inputs]( in ConcatCPUImpl()
88 for (size_t j = 0; j < num_inputs; ++j) { in ConcatCPUImpl()
114 inp.reserve(num_inputs); in ConcatCPUImpl()
120 for (int64 j = 0; j < num_inputs; ++j) { in ConcatCPUImpl()
141 size_t num_inputs = inputs.size(); in ConcatSYCLImpl() local
144 sizes.reserve(num_inputs); in ConcatSYCLImpl()
[all …]
Dsearchsorted_op.cc38 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()
Dstring_format_op.cc35 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()
Dsearchsorted_op_gpu.cu.cc66 int batch_size, int num_inputs, int num_values, in Compute()
74 batch_size, num_inputs, num_values, values.data(), output->data())); in Compute()
85 int batch_size, int num_inputs, int num_values, in Compute()
93 batch_size, num_inputs, num_values, values.data(), output->data())); in Compute()
/external/mesa3d/src/gallium/drivers/svga/
Dsvga_link.c62 assert(inshader_info->num_inputs <= in svga_link_shaders()
68 for (i = 0; i < inshader_info->num_inputs; i++) { in svga_link_shaders()
108 linkage->num_inputs = inshader_info->num_inputs; in svga_link_shaders()
112 for (i = 0; i < inshader_info->num_inputs; i++) { in svga_link_shaders()
128 linkage->num_inputs, linkage->input_map_max, in svga_link_shaders()
131 for (i = 0; i < linkage->num_inputs; i++) { in svga_link_shaders()
/external/tensorflow/tensorflow/compiler/tests/
Dlstm.py43 def LSTMCellWeightsShape(num_inputs, num_nodes): argument
47 return [num_inputs + num_nodes, 4 * num_nodes]
127 def RandomInputs(batch_size, seq_length, num_inputs): argument
133 x_seq.append(RandomVar([batch_size, num_inputs], name='x_seq_%d' % seq))
140 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
155 LSTMCellWeightsShape(num_inputs, num_nodes), name='weights')
158 x_seq, pad_seq = RandomInputs(batch_size, seq_length, num_inputs)
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc264 const int num_inputs = node.input_size(); in HasControlInputs() local
265 if (num_inputs > 0 && IsControlInput(node.input(num_inputs - 1))) { in HasControlInputs()
272 const int num_inputs = node.input_size(); in HasRegularInputs() local
273 if (num_inputs > 0 && !IsControlInput(node.input(0))) { in HasRegularInputs()
280 int num_inputs = 0; in NumNonControlInputs() local
281 for (; num_inputs < node.input_size(); ++num_inputs) { in NumNonControlInputs()
282 const string& input = node.input(num_inputs); in NumNonControlInputs()
284 return num_inputs; in NumNonControlInputs()
287 return num_inputs; in NumNonControlInputs()
291 int num_inputs = 0; in NumControlInputs() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dadd_n.cc30 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()
Dwhile.cc126 int num_inputs = node->inputs->size; in Prepare() local
128 TF_LITE_ENSURE_EQ(context, node->outputs->size, num_inputs); in Prepare()
140 TF_LITE_ENSURE_EQ(context, cond_subgraph->inputs().size(), num_inputs); in Prepare()
144 TF_LITE_ENSURE_EQ(context, body_subgraph->inputs().size(), num_inputs); in Prepare()
145 TF_LITE_ENSURE_EQ(context, body_subgraph->outputs().size(), num_inputs); in Prepare()
174 for (int i = 0; i < num_inputs; ++i) { in Prepare()
194 for (int i = 0; i < num_inputs; ++i) { in Prepare()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc55 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
68 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
96 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
197 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in Initialize() local
198 i < num_inputs; ++i) { in Initialize()
308 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in StartThreads() local
309 i < num_inputs; ++i) { in StartThreads()
338 for (size_t i = 1, num_inputs = histograms_.size(); i < num_inputs; in SelectFastestInputIndex() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dremove_nodes.cc55 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/compiler/tf2xla/kernels/
Dfunction_ops.cc32 OP_REQUIRES(ctx, ctx->num_inputs() == ctx->num_outputs(), in PassOn()
33 errors::Internal("#inputs != #outputs : ", ctx->num_inputs(), in PassOn()
35 for (int i = 0; i < ctx->num_inputs(); ++i) { in PassOn()
45 for (int i = 0; i < ctx->num_inputs(); ++i) { in Compile()
Daggregate_ops.cc32 OP_REQUIRES(ctx, ctx->num_inputs() >= 1, in Compile()
40 for (int i = 1; i < ctx->num_inputs(); ++i) { in Compile()
60 for (int i = 1; i < ctx->num_inputs(); ++i) { in Compile()
83 for (int i = 1; i < ctx->num_inputs(); ++i) { in Compile()
Dwhile_op.cc70 VLOG(2) << "Num inputs " << ctx->num_inputs(); in MakeXlaCompilerArgumentsFromInputs()
71 args->resize(ctx->num_inputs()); in MakeXlaCompilerArgumentsFromInputs()
75 for (int i = 0; i < ctx->num_inputs(); ++i) { in MakeXlaCompilerArgumentsFromInputs()
140 std::vector<bool> loop_invariants(ctx->num_inputs()); in ConvertLoopInvariantsToConst()
186 body.xla_output_shape.tuple_shapes(ctx->num_inputs()); in VerifyBodyInputAndOutputShapeMatch()
253 for (int i = 0; i < ctx->num_inputs(); i++) { in BuildWhile()
265 while_result, ctx->num_inputs() - num_compile_time_const_args)); in BuildWhile()
317 std::vector<bool> compile_time_const_arg_indices(ctx->num_inputs()); in Compile()
485 int num_inputs = body.input_mapping.size(); in Compile() local
486 std::vector<xla::XlaOp> inputs(num_inputs); in Compile()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor.cc112 kernel_state.num_inputs = n->num_inputs(); in Initialize()
121 previous_kernel_state.num_inputs; in Initialize()
182 last_kernel_state.input_start_index + last_kernel_state.num_inputs; in Initialize()
304 const size_t num_inputs = kernel_state.num_inputs; in Run() local
308 node_inputs.resize(num_inputs); in Run()
310 input_alloc_attrs.resize(num_inputs); in Run()
311 for (size_t j = 0; j < num_inputs; ++j) { in Run()
355 for (size_t j = 0; j < num_inputs; ++j) { in Run()
406 size_t num_inputs; member
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.cc82 const tensorflow::DataType dtype, int num_inputs) { in InferSingleTypeInputListAttrs() argument
85 MutableAttrs()->Set(input_def.number_attr(), num_inputs); in InferSingleTypeInputListAttrs()
106 tensorflow::Status EagerOperation::InferInputListAttrs(int num_inputs) { in InferInputListAttrs() argument
112 std::vector<tensorflow::DataType> dtypes(num_inputs); in InferInputListAttrs()
113 for (int i = 0; i < num_inputs; ++i) { in InferInputListAttrs()
119 InferSingleTypeInputListAttrs(input_def, inputs_[start]->dtype, num_inputs); in InferInputListAttrs()
/external/tensorflow/tensorflow/core/common_runtime/
Dconstant_folding_test.cc129 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 …]
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_concat_ops.py253 num_inputs = len(rt_inputs)
272 row_indices = math_ops.range(rt_nrows * num_inputs)
273 row_index_matrix = array_ops.reshape(row_indices, [num_inputs, -1])
280 stack_splits = math_ops.range(0, rt_nrows * num_inputs + 1, num_inputs)
287 concat_splits = permuted_rt.row_splits[::num_inputs]
/external/tensorflow/tensorflow/compiler/tf2tensorrt/plugin/
Dplugin_cast.cu.cc81 int num_inputs) const override { in getOutputDataType()
83 DCHECK_EQ(1, num_inputs); in getOutputDataType()
89 int num_inputs) const override { in isOutputBroadcastAcrossBatch()
97 void configurePlugin(const Dims* input_dims, int num_inputs, in configurePlugin() argument
104 DCHECK_EQ(1, num_inputs); in configurePlugin()
/external/tensorflow/tensorflow/core/ops/
Dmkl_array_ops.cc50 const int n = (c->num_inputs() - 1) / 3; in __anon42671a800102()
53 for (int i = n + 1; i < c->num_inputs(); ++i) { in __anon42671a800102()
80 const int n = (c->num_inputs() / 2 - 1) / 3; in __anon42671a800202()
83 for (int i = n + 1; i < c->num_inputs() / 2; ++i) { in __anon42671a800202()
/external/tensorflow/tensorflow/core/graph/
Doptimizer_cse.cc74 DCHECK_EQ(in->size(), n->num_inputs()); in FillInputs()
207 hasher.MixInteger(n->num_inputs()); in NodeHash()
209 gtl::InlinedVector<std::pair<const Node*, int>, 4> in(n->num_inputs()); in NodeHash()
256 if (a->num_inputs() != b->num_inputs()) return false; in Equivalent()
257 const int N_in = a->num_inputs(); in Equivalent()

12345678910>>...17