Searched refs:gather (Results 1 - 25 of 134) sorted by relevance

123456

/external/tensorflow/tensorflow/python/kernel_tests/
H A Dgather_op_test.py15 """Tests for tensorflow.ops.tf.gather."""
51 gather_t = array_ops.gather(params, indices_tf)
66 gather_t = array_ops.gather(params, indices, axis=axis)
82 gather_t = array_ops.gather(params, indices, axis=axis)
103 gather = array_ops.gather(tf_params, tf_indices, axis=tf_axis)
104 gather_negative_axis = array_ops.gather(
107 [gather, gather_negative_axis])
113 self.assertEqual(expected_shape, gather.shape)
118 *gather
[all...]
/external/tensorflow/tensorflow/compiler/tests/
H A Dgather_test.py52 gather_t = array_ops.gather(params, indices_tf)
66 gather_t = array_ops.gather(params, indices, axis=axis)
81 gather_t = array_ops.gather(params, indices, axis=axis)
100 gather_t = array_ops.gather(params, indices, axis=axis)
120 gather = array_ops.gather(tf_params, tf_indices, axis=axis)
121 gather_value = sess.run(gather, feed_dict={tf_params: params})
131 gather = array_ops.gather(params, indices)
133 7, gather
[all...]
/external/tensorflow/tensorflow/contrib/opt/python/training/
H A Dlazy_adam_optimizer.py62 beta1_t * array_ops.gather(m, grad.indices) +
69 beta2_t * array_ops.gather(v, grad.indices) +
74 m_t_slice = array_ops.gather(m_t, grad.indices)
75 v_t_slice = array_ops.gather(v_t, grad.indices)
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
H A Dpermute.py126 return array_ops.gather(x, self.permutation, axis=-1)
129 return array_ops.gather(
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dgather_op.cc106 // Construct the while loop body's function. The implementation of gather is:
159 init, "gather", builder));
198 xla::ComputationDataHandle gather; variable
202 builder, &gather));
203 context->SetOutput(0, gather);
239 xla::ComputationDataHandle gather; variable
243 indices_type, builder, &gather));
244 context->SetOutput(0, gather);
H A Dvariable_ops.cc120 xla::ComputationDataHandle gather; variable
124 builder, &gather));
125 ctx->SetOutput(0, gather);
/external/tensorflow/tensorflow/compiler/aot/tests/
H A Dtfcompile_test.cc146 GatherComp gather; local
147 EXPECT_EQ(gather.arg0_data(), gather.args()[0]);
148 EXPECT_EQ(gather.arg1_data(), gather.args()[1]);
150 // Successful gather.
153 std::copy(params + 0, params + 4, gather.arg0_data());
155 std::copy(indices + 0, indices + 2, gather.arg1_data());
156 EXPECT_TRUE(gather.Run());
157 EXPECT_EQ(gather
[all...]
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
H A Dcategorical_split_handler.py124 filtered_gradients = array_ops.gather(gradients, example_indices)
125 filtered_hessians = array_ops.gather(hessians, example_indices)
126 filtered_partition_ids = array_ops.gather(example_partition_ids,
/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dgather.cc27 namespace gather { namespace in namespace:tflite::ops::builtin
121 } // namespace gather
124 static TfLiteRegistration r = {nullptr, nullptr, gather::Prepare,
125 gather::Eval};
/external/tensorflow/tensorflow/contrib/nn/python/ops/
H A Dfwd_gradients_test.py43 y = array_ops.gather(x, [0, 1])
H A Dsampling_ops.py103 resampled = array_ops.gather(sampled, indices=resampled_indices)
104 resampled_expected_count = array_ops.gather(
/external/skia/src/gpu/
H A DGrTextureOpList.cpp164 auto gather = [ alloc SkDEBUGCODE(, this) ] (GrSurfaceProxy* p) { local
170 op->visitProxies(gather);
/external/skqp/src/gpu/
H A DGrTextureOpList.cpp164 auto gather = [ alloc SkDEBUGCODE(, this) ] (GrSurfaceProxy* p) { local
170 op->visitProxies(gather);
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
H A Dfactorization_ops_test_utils.py128 gathered_row_factors = array_ops.gather(row_factors, row_indices)
129 gathered_col_factors = array_ops.gather(col_factors, col_indices)
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/
H A Dconditional_transformed_distribution_test.py61 return array_ops.gather(self._loc, z)
/external/tensorflow/tensorflow/python/ops/
H A Dsparse_grad.py61 array_ops.gather(output_values_grad, inverted_permutation),
165 # TODO(zongheng): these gather calls could potentially duplicate rows/cols in
168 parts_a = array_ops.gather(grad, rows if not adj_a else cols)
169 parts_b = array_ops.gather(b if not adj_b else array_ops.transpose(b),
/external/skia/src/jumper/
H A DSkJumper_stages.cpp83 SI T gather(const T* p, U32 ix) { return p[ix]; } function
165 SI V<T> gather(const T* p, U32 ix) { function
270 SI V<T> gather(const T* p, U32 ix) { function
275 SI F gather(const float* p, U32 ix) { return _mm256_i32gather_ps (p, ix, 4); } function
276 SI U32 gather(const uint32_t* p, U32 ix) { return _mm256_i32gather_epi32(p, ix, 4); } function
277 SI U64 gather(const uint64_t* p, U32 ix) { function
494 SI V<T> gather(const T* p, U32 ix) { function
1467 r = gather(c->r, (px ) & 0xff);
1468 g = gather(c->g, (px >> 8) & 0xff);
1469 b = gather(
[all...]
H A DSkJumper_stages_lowp.cpp80 // GP: geometry in, pixels out. -- think, a memory gather
470 SI V gather(const T* ptr, U32 ix) { function
478 F gather(const float* p, U32 ix) { function
487 U32 gather(const uint32_t* p, U32 ix) { function
496 SI V gather(const T* ptr, U32 ix) { function
598 from_8888(gather<U32>(ptr, ix), &r, &g, &b, &a);
603 from_8888(gather<U32>(ptr, ix), &b, &g, &r, &a);
647 from_565(gather<U16>(ptr, ix), &r, &g, &b);
692 from_4444(gather<U16>(ptr, ix), &r,&g,&b,&a);
719 a = cast<U16>(gather<U
[all...]
/external/skqp/src/jumper/
H A DSkJumper_stages.cpp83 SI T gather(const T* p, U32 ix) { return p[ix]; } function
165 SI V<T> gather(const T* p, U32 ix) { function
270 SI V<T> gather(const T* p, U32 ix) { function
275 SI F gather(const float* p, U32 ix) { return _mm256_i32gather_ps (p, ix, 4); } function
276 SI U32 gather(const uint32_t* p, U32 ix) { return _mm256_i32gather_epi32(p, ix, 4); } function
277 SI U64 gather(const uint64_t* p, U32 ix) { function
494 SI V<T> gather(const T* p, U32 ix) { function
1465 r = gather(c->r, (px ) & 0xff);
1466 g = gather(c->g, (px >> 8) & 0xff);
1467 b = gather(
[all...]
H A DSkJumper_stages_lowp.cpp80 // GP: geometry in, pixels out. -- think, a memory gather
470 SI V gather(const T* ptr, U32 ix) { function
478 F gather(const float* p, U32 ix) { function
487 U32 gather(const uint32_t* p, U32 ix) { function
496 SI V gather(const T* ptr, U32 ix) { function
598 from_8888(gather<U32>(ptr, ix), &r, &g, &b, &a);
603 from_8888(gather<U32>(ptr, ix), &b, &g, &r, &a);
647 from_565(gather<U16>(ptr, ix), &r, &g, &b);
692 from_4444(gather<U16>(ptr, ix), &r,&g,&b,&a);
719 a = cast<U16>(gather<U
[all...]
/external/skia/src/shaders/
H A DSkImageShader.cpp355 auto gather = alloc->make<SkJumper_GatherCtx>(); local
356 gather->pixels = pm.addr();
357 gather->stride = pm.rowBytesAsPixels();
358 gather->width = pm.width();
359 gather->height = pm.height();
396 void* ctx = gather;
447 p->append(SkRasterPipeline::bilerp_clamp_8888, gather);
/external/skqp/src/shaders/
H A DSkImageShader.cpp315 auto gather = alloc->make<SkJumper_GatherCtx>(); local
316 gather->pixels = pm.addr();
317 gather->stride = pm.rowBytesAsPixels();
318 gather->width = pm.width();
319 gather->height = pm.height();
341 void* ctx = gather;
389 p->append(SkRasterPipeline::bilerp_clamp_8888, gather);
/external/llvm/lib/Transforms/Scalar/
H A DScalarizer.cpp164 void gather(Instruction *, const ValueVector &);
296 void Scalarizer::gather(Instruction *Op, const ValueVector &CV) { function in class:Scalarizer
393 gather(&I, Res);
423 gather(&SI, Res);
468 gather(&GEPI, Res);
486 gather(&CI, Res);
541 gather(&BCI, Res);
565 gather(&SVI, Res);
590 gather(&PHI, Res);
614 gather(
[all...]
/external/python/cpython2/Lib/test/
H A Dtest_sax.py735 gather = self.AttrGatherer()
736 parser.setContentHandler(gather)
741 self.verify_empty_attrs(gather._attrs)
745 gather = self.AttrGatherer()
746 parser.setContentHandler(gather)
751 self.verify_attrs_wattr(gather._attrs)
755 gather = self.AttrGatherer()
756 parser.setContentHandler(gather)
761 self.verify_empty_nsattrs(gather._attrs)
765 gather
[all...]
/external/python/cpython3/Lib/test/
H A Dtest_sax.py954 gather = self.AttrGatherer()
955 parser.setContentHandler(gather)
960 self.verify_empty_attrs(gather._attrs)
964 gather = self.AttrGatherer()
965 parser.setContentHandler(gather)
970 self.verify_attrs_wattr(gather._attrs)
974 gather = self.AttrGatherer()
975 parser.setContentHandler(gather)
980 self.verify_empty_nsattrs(gather._attrs)
984 gather
[all...]

Completed in 1229 milliseconds

123456