| /external/mesa3d/src/gallium/drivers/nouveau/ci/ |
| D | nouveau-gt215-fails.txt | 127 # Gather with non-zero comp doesn't work 129 KHR-GLES31.core.texture_gather.offset-gather-depth-2d,Fail 130 KHR-GLES31.core.texture_gather.offset-gather-depth-2darray,Fail 131 KHR-GLES31.core.texture_gather.offset-gather-int-2darray,Fail 132 KHR-GLES31.core.texture_gather.offset-gather-uint-2d,Fail 133 KHR-GLES31.core.texture_gather.plain-gather-depth-2d,Fail 134 KHR-GLES31.core.texture_gather.plain-gather-depth-2darray,Fail 135 KHR-GLES31.core.texture_gather.plain-gather-depth-cube,Fail 136 KHR-GLES31.core.texture_gather.plain-gather-float-2d-srgb-alpha,Fail 137 KHR-GLES31.core.texture_gather.plain-gather-int-2darray,Fail [all …]
|
| /external/mesa3d/src/panfrost/ci/ |
| D | panfrost-t760-fails.txt | 42 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.compressed_color.gather.s… 43 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.compressed_color.gather.s… 44 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.float_depth.gather.s_mirr… 45 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.float_depth.gather.s_mirr… 46 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.snorm_color.gather.s_mirr… 47 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.snorm_color.gather.s_mirr… 48 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.unorm_color.gather.s_mirr… 49 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.unorm_color.gather.s_mirr… 50 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.unorm_depth.gather.s_mirr… 51 dEQP-GLES31.functional.texture.border_clamp.per_axis_wrap_mode.texture_2d.unorm_depth.gather.s_mirr… [all …]
|
| /external/mesa3d/src/gallium/drivers/softpipe/ci/ |
| D | softpipe-asan-fails.txt | 31 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8.no_corners.size_pot.mirrored_repeat_clamp_to… 32 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8.no_corners.size_pot.repeat_mirrored_repeat,F… 33 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8i.no_corners.size_pot.mirrored_repeat_clamp_t… 34 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8ui.no_corners.size_pot.repeat_mirrored_repeat… 35 dEQP-GLES31.functional.texture.gather.offset_dynamic.implementation_offset.2d.depth32f.base_level.l… 36 dEQP-GLES31.functional.texture.gather.offset_dynamic.implementation_offset.2d.depth32f.size_npot.co… 37 dEQP-GLES31.functional.texture.gather.offset_dynamic.implementation_offset.2d.depth32f.size_npot.co… 38 dEQP-GLES31.functional.texture.gather.offset_dynamic.implementation_offset.2d.depth32f.size_npot.co… 39 dEQP-GLES31.functional.texture.gather.offset_dynamic.implementation_offset.2d.depth32f.size_pot.com… 40 dEQP-GLES31.functional.texture.gather.offset_dynamic.implementation_offset.2d.rgba8.base_level.leve… [all …]
|
| D | softpipe-fails.txt | 509 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8.no_corners.size_pot.clamp_to_edge_repeat,Fail 510 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8.no_corners.size_pot.mirrored_repeat_clamp_to… 511 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8.no_corners.size_pot.repeat_mirrored_repeat,F… 512 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8i.no_corners.size_pot.clamp_to_edge_repeat,Fa… 513 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8i.no_corners.size_pot.mirrored_repeat_clamp_t… 514 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8i.no_corners.size_pot.repeat_mirrored_repeat,… 515 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8ui.no_corners.size_pot.clamp_to_edge_repeat,F… 516 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8ui.no_corners.size_pot.mirrored_repeat_clamp_… 517 dEQP-GLES31.functional.texture.gather.basic.cube.rgba8ui.no_corners.size_pot.repeat_mirrored_repeat… 518 dEQP-GLES31.functional.texture.gather.offset_dynamic.implementation_offset.2d.depth32f.base_level.l… [all …]
|
| /external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/aosp_mustpass/3.2.4.x/src/ |
| D | gles31-hw-issues.txt | 2 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.clamp_to_edge_repeat 3 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.repeat_mirrored_rep… 4 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.mirrored_repeat_cla… 5 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.clamp_to_edge_re… 6 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.repeat_mirrored_… 7 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.mirrored_repeat_… 8 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.clamp_to_edge_repe… 9 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.repeat_mirrored_re… 10 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.mirrored_repeat_cl… 11 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_greater.clamp_to_edge_r… [all …]
|
| /external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/aosp_mustpass/main/src/ |
| D | gles31-hw-issues.txt | 2 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.clamp_to_edge_repeat 3 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.repeat_mirrored_rep… 4 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.mirrored_repeat_cla… 5 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.clamp_to_edge_re… 6 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.repeat_mirrored_… 7 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.mirrored_repeat_… 8 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.clamp_to_edge_repe… 9 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.repeat_mirrored_re… 10 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.mirrored_repeat_cl… 11 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_greater.clamp_to_edge_r… [all …]
|
| /external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/aosp_mustpass/3.2.5.x/src/ |
| D | gles31-hw-issues.txt | 2 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.clamp_to_edge_repeat 3 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.repeat_mirrored_rep… 4 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.mirrored_repeat_cla… 5 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.clamp_to_edge_re… 6 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.repeat_mirrored_… 7 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.mirrored_repeat_… 8 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.clamp_to_edge_repe… 9 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.repeat_mirrored_re… 10 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.mirrored_repeat_cl… 11 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_greater.clamp_to_edge_r… [all …]
|
| /external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/aosp_mustpass/3.2.6.x/src/ |
| D | gles31-hw-issues.txt | 2 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.clamp_to_edge_repeat 3 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.repeat_mirrored_rep… 4 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.mirrored_repeat_cla… 5 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.clamp_to_edge_re… 6 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.repeat_mirrored_… 7 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.mirrored_repeat_… 8 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.clamp_to_edge_repe… 9 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.repeat_mirrored_re… 10 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.mirrored_repeat_cl… 11 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_greater.clamp_to_edge_r… [all …]
|
| /external/deqp/external/openglcts/data/gl_cts/data/mustpass/gles/aosp_mustpass/3.2.3.x/src/ |
| D | gles31-hw-issues.txt | 2 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.clamp_to_edge_repeat 3 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.repeat_mirrored_rep… 4 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_less.mirrored_repeat_cla… 5 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.clamp_to_edge_re… 6 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.repeat_mirrored_… 7 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_pot.compare_greater.mirrored_repeat_… 8 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.clamp_to_edge_repe… 9 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.repeat_mirrored_re… 10 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_less.mirrored_repeat_cl… 11 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.size_npot.compare_greater.clamp_to_edge_r… [all …]
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | array_test.py | 41 x_i = array_ops.gather(x, i) 44 outputs.append(array_ops.gather(y, 2, axis=axis)) 46 array_ops.gather(y, math_ops.cast(2, dtypes.int64), axis=axis)) 48 array_ops.gather(y, 2, axis=math_ops.cast(axis, dtypes.int64))) 50 array_ops.gather(y, math_ops.cast(i, dtypes.int64), axis=axis)) 51 outputs.append(array_ops.gather(y, [i], axis=axis)) 52 outputs.append(array_ops.gather(y, [i, 2], axis=axis)) 53 outputs.append(array_ops.gather(y, [[2, i], [i, 1]], axis=axis)) 55 outputs.append(array_ops.gather(y, [0, 1, 2], axis=1, batch_dims=1)) 56 outputs.append(array_ops.gather(y, [i, 1, 2], axis=2, batch_dims=1)) [all …]
|
| D | math_test.py | 55 x_i = array_ops.gather(x, i) 144 return op(array_ops.gather(x, i)) 216 x1 = array_ops.gather(x, i) 217 y1 = array_ops.gather(y, i) 235 x1 = array_ops.gather(x, i) 236 y1 = array_ops.gather(y, i) 250 x1 = array_ops.gather(a, i) 262 x1 = array_ops.gather(x, i) 272 x_i = array_ops.gather(x, i) 273 y_i = array_ops.gather(y, i) [all …]
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | optimize_mhlo.cc | 55 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite() argument 57 auto dimensionNumbers = gather.dimension_numbers(); in matchAndRewrite() 60 if (!gather.operand().getType().cast<ShapedType>().hasRank() || in matchAndRewrite() 61 !gather.operand().getType().cast<ShapedType>().hasStaticShape() || in matchAndRewrite() 62 !gather.start_indices().getType().cast<ShapedType>().hasRank() || in matchAndRewrite() 63 !gather.start_indices().getType().cast<ShapedType>().hasStaticShape()) { in matchAndRewrite() 64 return rewriter.notifyMatchFailure(gather, in matchAndRewrite() 69 return rewriter.notifyMatchFailure(gather, "non-zero index_vector_dim"); in matchAndRewrite() 76 return rewriter.notifyMatchFailure(gather, in matchAndRewrite() 80 auto resultTy = gather.getResult().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() [all …]
|
| D | legalize_gather_to_torch_index_select.cc | 33 LogicalResult matchAndRewrite(GatherOp gather, in matchAndRewrite() 35 auto startIndices = gather.start_indices(); in matchAndRewrite() 38 return rewriter.notifyMatchFailure(gather, "unranked start_indices"); in matchAndRewrite() 41 auto operand = gather.operand(); in matchAndRewrite() 44 return rewriter.notifyMatchFailure(gather, "unranked operand"); in matchAndRewrite() 50 // gather indices. in matchAndRewrite() 51 auto dimensionNumbers = gather.dimension_numbers(); in matchAndRewrite() 54 gather, "index_vector_dim not last dimension of start_indices"); in matchAndRewrite() 61 gather, "start_indices index vector dimension not 1"); in matchAndRewrite() 67 return rewriter.notifyMatchFailure(gather, "start_index_map != [0]"); in matchAndRewrite() [all …]
|
| /external/deqp/android/cts/main/src/ |
| D | gles31-hw-issues.txt | 35 dEQP-GLES31.functional.texture.gather.basic.2d.depth32f.size_pot.compare_greater.mirrored_repeat_cl… 36 dEQP-GLES31.functional.texture.gather.basic.2d.depth32f.size_pot.compare_greater.repeat_mirrored_re… 37 dEQP-GLES31.functional.texture.gather.basic.2d.depth32f.size_pot.compare_less.mirrored_repeat_clamp… 38 dEQP-GLES31.functional.texture.gather.basic.2d.depth32f.size_pot.compare_less.repeat_mirrored_repeat 41 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.base_level.level_1 42 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.base_level.level_2 43 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.filter_mode.min_linear_mag_linear 44 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.filter_mode.min_linear_mipmap_linear_mag_… 45 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.filter_mode.min_linear_mipmap_nearest_mag… 46 dEQP-GLES31.functional.texture.gather.basic.cube.depth32f.filter_mode.min_nearest_mipmap_linear_mag… [all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_simplifier_test.cc | 38 ROOT gather = f32[42,43,7,8] gather(operand, indices), in TEST_F() 49 CHECK: %[[GATHER:.*]] = f32[1806,7,8]{{.*}} gather( in TEST_F() 56 CHECK: ROOT %{{.*}} = f32[42,43,7,8]{3,2,1,0} reshape(%[[GATHER]]) in TEST_F() 69 ROOT gather = f32[42] gather(operand, indices), in TEST_F() 78 CHECK: %[[GATHER:.*]] = f32[42,1,1]{2,1,0} gather(%operand, %indices) in TEST_F() 81 CHECK: ROOT %{{.*}} = f32[42]{0} reshape(%[[GATHER]]) in TEST_F() 93 ROOT gather = f32[42,1,2,3] gather(operand, indices), in TEST_F() 104 CHECK: %[[GATHER:.*]] = f32[42,3,1,2]{{.*}} gather(%[[OPERAND]], in TEST_F() 106 CHECK: ROOT {{.*}} = f32[42,1,2,3]{{.*}} transpose(%[[GATHER]]) in TEST_F() 118 ROOT gather = f32[7,42] gather(operand, indices), in TEST_F() [all …]
|
| D | indexed_array_analysis_test.cc | 84 ROOT gather = s32[5,3] gather(operand, indices), in TEST_F() 104 ROOT gather = s32[5,3] gather(operand, indices), in TEST_F() 124 ROOT gather = s32[5] gather(operand, indices), in TEST_F() 133 AssertArrayForRootExpressionIs(hlo_text, "%gather"); in TEST_F() 143 ROOT gather = s32[5,3] gather(operand, indices), in TEST_F() 152 AssertArrayForRootExpressionIs(hlo_text, "%gather"); in TEST_F() 162 ROOT gather = s32[5,2,3] gather(operand, indices), in TEST_F() 171 AssertArrayForRootExpressionIs(hlo_text, "%gather"); in TEST_F() 181 ROOT gather = s32[5,2] gather(operand, indices), in TEST_F() 190 AssertArrayForRootExpressionIs(hlo_text, "%gather"); in TEST_F() [all …]
|
| D | gather_expander_test.cc | 35 ROOT gather = s32[2147483647,3,5] gather(operand, indices), in TEST_F() 53 ::testing::HasSubstr("Gather operations with more than 2147483647 gather " in TEST_F() 64 ROOT gather = s32[3,2] gather(operand, indices), in TEST_F() 84 "after gather expansion"; in TEST_F() 91 "after gather expansion"; in TEST_F() 94 // dimensions for TF gather. In this case we expect the loop state to be of in TEST_F() 124 ROOT gather = s32[3,2] gather(operand, indices), in TEST_F() 135 metadata.set_op_name("Gather"); in TEST_F() 147 "after gather expansion"; in TEST_F() 154 "after gather expansion"; in TEST_F() [all …]
|
| D | all_gather_broadcast_reorder.cc | 36 "all-gather with constrained layouts"; in Run() 45 // Check for all-gather with a broadcast operand. in Run() 55 // We categorize each dimension of the all-gather result as either in Run() 58 // the all-gather participants. In addition, the set of dimensions in Run() 59 // for the broadcast instruction feeding into all-gather are also not in Run() 61 // then doing the broadcast after the all-gather is beneficial as the in Run() 62 // all-gather will be operating on smaller data. in Run() 83 // Transform the all-gather(broadcast(x)) to broadcast(all-gather(x)). in Run() 86 // If the all-gather happens along one of the non-uniform dimensions of in Run() 87 // the broadcast, then issue all-gather(x) and then a broadcast. in Run() [all …]
|
| D | gather_simplifier.cc | 35 auto* gather = DynCast<HloGatherInstruction>(inst); in ExpandInstruction() local 38 if (absl::c_linear_search(gather->gather_slice_sizes(), 0)) { in ExpandInstruction() 39 auto* zero = gather->AddInstruction(HloInstruction::CreateConstant( in ExpandInstruction() 40 LiteralUtil::Zero(gather->shape().element_type()))); in ExpandInstruction() 41 return gather->AddInstruction( in ExpandInstruction() 42 HloInstruction::CreateBroadcast(gather->shape(), zero, {})); in ExpandInstruction() 45 const auto& dims = gather->gather_dimension_numbers(); in ExpandInstruction() 52 auto* operand = gather->operands()[0]; in ExpandInstruction() 53 auto* start_indices = gather->operands()[1]; in ExpandInstruction() 60 // output shape for the Gather op. in ExpandInstruction() [all …]
|
| /external/pytorch/docs/source/notes/ |
| D | fsdp.rst | 17 stream to allow for overlapping an all-gather with ``forward`` compute issued before it (from the C… 18 perspective). For example, if we have layer 0 all-gather -> layer 0 ``forward`` compute -> layer 1 19 all-gather -> …, then layer 1 all-gather can overlap with layer 0 ``forward`` compute even though t… 20 CPU thread issued it afterwards. (The 1st all-gather will not be able to overlap with anything.) 23 all-gather -> layer 1 all-gather -> layer 0 ``forward`` compute -> …. In eager mode, there is no wa… 31 the cost that the next all-gather’s output tensor must be allocated while the current one is still 32 in use. By issuing the next all- gather before the current ``forward`` compute kernels, the next 33 all-gather can start sooner on GPU. For most LLM workloads, this is not the case, so there is no 38 all-gather and reduce-scatter (partially because in earlier NCCL versions, it was not safe to use 41 we explicitly reorder the CPU issue order to be next all-gather -> current reduce-scatter, then the [all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | gather_operation_test.cc | 53 ROOT gather = s32[2,3] gather(operand, indices), in XLA_TEST_F() 75 ROOT gather = s32[3, 1, 2, 3] gather(operand, indices), in XLA_TEST_F() 98 ROOT gather = s32[3,2] gather(operand, indices), in XLA_TEST_F() 119 ROOT gather = s32[2,3,2] gather(operand, indices), in XLA_TEST_F() 140 ROOT gather = s32[2,2] gather(operand, indices), in XLA_TEST_F() 162 ROOT gather = s32[2,1,1,2] gather(operand, indices), in XLA_TEST_F() 184 ROOT gather = s32[2,2] gather(operand, indices), in XLA_TEST_F() 207 ROOT gather = s32[2,2] gather(operand, indices), in XLA_TEST_F() 230 ROOT gather = s32[1,1] gather(operand, indices), in XLA_TEST_F() 251 ROOT gather = s32[2,1,1] gather(operand, indices), in XLA_TEST_F() [all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | gather_op_test.py | 15 """Tests for tensorflow.ops.tf.gather.""" 71 gather_t = array_ops.gather(params, indices_tf) 87 gather_t = array_ops.gather(params, indices, axis=axis) 104 gather_t = array_ops.gather(params, indices, axis=axis) 130 gather = array_ops.gather(tf_params, tf_indices, axis=tf_axis) 131 gather_negative_axis = array_ops.gather( 134 [gather, gather_negative_axis]) 140 self.assertEqual(expected_shape, gather.shape) 145 *gather.get_shape().as_list()).astype(dtype.as_numpy_dtype) 149 gather, [tf_params, tf_indices, tf_axis], gather_grad) [all …]
|
| /external/sdv/vsomeip/third_party/boost/algorithm/doc/ |
| D | gather.qbk | 1 [/ File gather.qbk] 3 [section:gather gather] 12 The header file 'boost/algorithm/gather.hpp' contains two variants of a single algorithm, `gather`. 14 `gather()` takes a collection of elements defined by a pair of iterators and moves the ones satisfy… 18 The function `gather` returns a `std::pair` of iterators that denote the elements that satisfy the … 27 gather ( BidirectionalIterator first, BidirectionalIterator last, BidirectionalIterator pivot, Pred… 31 gather ( const BidirectionalRange &range, typename boost::range_iterator<const BidirectionalRange>:… 43 a call to gather ( arr, arr + 10, arr + 4, IsEven ) will result in: 56 …gather` work on bidirectional iterators or better. This requirement comes from the usage of `stabl… 60 `gather` uses `stable_partition`, which will attempt to allocate temporary memory, but will work in… [all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather.cc | 30 namespace gather { namespace 56 "Positions of type '%s' are not supported by gather.", in Prepare() 79 TF_LITE_KERNEL_LOG(context, "Type '%s' is not supported by gather.", in Prepare() 120 TfLiteStatus Gather(TfLiteContext* context, const TfLiteGatherParams& params, in Gather() function 137 optimized_ops::Gather(op_params, GetTensorShape(input), in Gather() 189 return Gather<float, int32_t>(context, *params, input, positions, in Eval() 192 return Gather<uint8_t, int32_t>(context, *params, input, positions, in Eval() 195 return Gather<int8_t, int32_t>(context, *params, input, positions, in Eval() 198 return Gather<int16_t, int32_t>(context, *params, input, positions, in Eval() 201 return Gather<int32_t, int32_t>(context, *params, input, positions, in Eval() [all …]
|
| /external/llvm/test/Analysis/CostModel/X86/ |
| D | masked-intrinsic-cost.ll | 73 ; AVX2: Found an estimated cost of 7 {{.*}}.gather 76 ; KNL: Found an estimated cost of 7 {{.*}}.gather 79 ; SKX: Found an estimated cost of 7 {{.*}}.gather 81 %res = call <2 x double> @llvm.masked.gather.v2f64(<2 x double*> %ptrs, i32 4, <2 x i1> %mask, <2 x… 84 declare <2 x double> @llvm.masked.gather.v2f64(<2 x double*> %ptrs, i32, <2 x i1> %mask, <2 x doubl… 89 ; AVX2: Found an estimated cost of 16 {{.*}}.gather 92 ; KNL: Found an estimated cost of 16 {{.*}}.gather 95 ; SKX: Found an estimated cost of 6 {{.*}}.gather 97 %res = call <4 x i32> @llvm.masked.gather.v4i32(<4 x i32*> %ptrs, i32 4, <4 x i1> %mask, <4 x i32> … 104 ; AVX2: Found an estimated cost of 8 {{.*}}.gather [all …]
|