/external/tensorflow/tensorflow/python/kernel_tests/ |
H A D | gather_op_test.py | 15 """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 D | gather_test.py | 52 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 D | lazy_adam_optimizer.py | 62 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 D | permute.py | 126 return array_ops.gather(x, self.permutation, axis=-1) 129 return array_ops.gather(
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | gather_op.cc | 106 // 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 D | variable_ops.cc | 120 xla::ComputationDataHandle gather; variable 124 builder, &gather)); 125 ctx->SetOutput(0, gather);
|
/external/tensorflow/tensorflow/compiler/aot/tests/ |
H A D | tfcompile_test.cc | 146 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 D | categorical_split_handler.py | 124 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 D | gather.cc | 27 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 D | fwd_gradients_test.py | 43 y = array_ops.gather(x, [0, 1])
|
H A D | sampling_ops.py | 103 resampled = array_ops.gather(sampled, indices=resampled_indices) 104 resampled_expected_count = array_ops.gather(
|
/external/skia/src/gpu/ |
H A D | GrTextureOpList.cpp | 164 auto gather = [ alloc SkDEBUGCODE(, this) ] (GrSurfaceProxy* p) { local 170 op->visitProxies(gather);
|
/external/skqp/src/gpu/ |
H A D | GrTextureOpList.cpp | 164 auto gather = [ alloc SkDEBUGCODE(, this) ] (GrSurfaceProxy* p) { local 170 op->visitProxies(gather);
|
/external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
H A D | factorization_ops_test_utils.py | 128 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 D | conditional_transformed_distribution_test.py | 61 return array_ops.gather(self._loc, z)
|
/external/tensorflow/tensorflow/python/ops/ |
H A D | sparse_grad.py | 61 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 D | SkJumper_stages.cpp | 83 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 D | SkJumper_stages_lowp.cpp | 80 // 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 D | SkJumper_stages.cpp | 83 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 D | SkJumper_stages_lowp.cpp | 80 // 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 D | SkImageShader.cpp | 355 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 D | SkImageShader.cpp | 315 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 D | Scalarizer.cpp | 164 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 D | test_sax.py | 735 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 D | test_sax.py | 954 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...] |