Home
last modified time | relevance | path

Searched refs:weights1 (Results 1 – 7 of 7) sorted by relevance

/external/libgav1/libgav1/src/dsp/x86/
Dintrapred_smooth_sse4.cc96 const __m128i& weights1, in WriteSmoothDirectionalSum16() argument
101 const __m128i weighted_px1 = _mm_mullo_epi16(pixels1, weights1); in WriteSmoothDirectionalSum16()
767 const __m128i weights1 = _mm_cvtepu8_epi16(weights); in SmoothHorizontal16x4_SSE4_1() local
769 const __m128i inverted_weights1 = _mm_sub_epi16(scale, weights1); in SmoothHorizontal16x4_SSE4_1()
779 WriteSmoothDirectionalSum16(dst, left_y, left_y, weights1, weights2, in SmoothHorizontal16x4_SSE4_1()
784 WriteSmoothDirectionalSum16(dst, left_y, left_y, weights1, weights2, in SmoothHorizontal16x4_SSE4_1()
789 WriteSmoothDirectionalSum16(dst, left_y, left_y, weights1, weights2, in SmoothHorizontal16x4_SSE4_1()
794 WriteSmoothDirectionalSum16(dst, left_y, left_y, weights1, weights2, in SmoothHorizontal16x4_SSE4_1()
806 const __m128i weights1 = _mm_cvtepu8_epi16(weights); in SmoothHorizontal16x8_SSE4_1() local
808 const __m128i inverted_weights1 = _mm_sub_epi16(scale, weights1); in SmoothHorizontal16x8_SSE4_1()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py351 weights1 = variables.Variable(
364 hidden1 = nn_ops.relu(math_ops.matmul(images, weights1) + biases1)
762 weights1 = variables.Variable([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]],
766 nn_ops.relu(math_ops.matmul(images, weights1) + biases1, name="relu")
789 weights1 = variables.Variable(
795 weights1 = variables.Variable(
797 nn_ops.relu(math_ops.matmul(images, weights1) + biases1, name="relu")
/external/tensorflow/tensorflow/python/keras/saving/
Dsave_weights_test.py206 weights1 = layer.get_weights()
208 layer, weights1)
211 for (x, y) in zip(weights1, weights2)
/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment_test.cc2071 auto weights1 = builder.AddInstruction( in TEST_F() local
2098 HloInstruction::CreateTuple({input1, weights1, output1})); in TEST_F()
2116 EXPECT_EQ(assignment->GetUniqueSlice(weights1, {}).ConsumeValueOrDie(), in TEST_F()
2575 auto weights1 = builder.AddInstruction( in TEST_F() local
2587 HloInstruction::CreateTuple({input1, weights1, output1})); in TEST_F()
2619 {input1, weights1, one, output1, while1->mutable_operand(0), while1, in TEST_F()
/external/tensorflow/tensorflow/python/training/
Dsaver_test.py2685 weights1 = variables.VariableV1(
2698 hidden1 = nn_ops.relu(math_ops.matmul(images, weights1) + biases1)
2768 weights1 = graph.as_graph_element("new_hidden1/weights:0")
2813 self.evaluate([weights1, biases1])
2837 weights1 = variables.VariableV1(
2840 nn_ops.relu(math_ops.matmul(images, weights1) + biases1, name="relu")
2894 weights1 = variables.VariableV1(
2897 nn_ops.relu(math_ops.matmul(images, weights1) + biases1, name="relu")
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h515 int8x16_t weights1 = vld1q_s8(shuffled_weights_ptr + 16); in ShuffledFullyConnectedWorkerImpl() local
524 vmull_s8(vget_low_s8(weights1), vget_low_s8(input)); in ShuffledFullyConnectedWorkerImpl()
532 vmlal_s8(local_accum1, vget_high_s8(weights1), vget_high_s8(input)); in ShuffledFullyConnectedWorkerImpl()
596 int8x16_t weights1 = vld1q_s8(shuffled_weights_ptr + 16); in ShuffledFullyConnectedWorkerImpl() local
608 local_accum1 = vmull_s8(vget_low_s8(weights1), vget_low_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
614 vmlal_s8(local_accum1, vget_high_s8(weights1), vget_high_s8(input##B)); \ in ShuffledFullyConnectedWorkerImpl()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc3059 Output weights1 = ops::Const(s.WithOpName("weights1"), in TEST_F() local
3069 ops::Conv2D(s.WithOpName("conv1"), input, weights1, {1, 1, 1, 1}, "SAME"); in TEST_F()