| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | select_and_scatter_test.cc | 45 std::vector<int64_t> source_shape; member
80 auto source_shape = GetParam().source_shape; in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | infeed_thunk.cc | 46 const Shape& source_shape = in ExecuteOnStream() local
|
| H A D | jitrt_custom_calls.cc | 986 const Shape& source_shape = in operator ()() local
1064 Shape source_shape = ToShape(*source); in operator ()() local
|
| H A D | ir_emitter_unnested.cc | 1903 const Shape source_shape = GetShape(select_and_scatter_op.getSource()); in EmitSelectAndScatter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | dynamic_ragged_shape.py | 2092 source_shape, argument
2142 def source_shape(self): member in _Broadcaster
2267 def _get_broadcaster(source_shape, target_shape): argument
|
| H A D | dynamic_ragged_shape_test.py | 2397 def testBroadcasterInitRaises(self, source_shape, target_shape, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/lazy/core/ops/ |
| H A D | utils.cpp | 49 const Shape& source_shape, in MakePermuteShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | select_and_scatter_expander.cc | 46 const auto source_shape = source->shape(); in ExpandInstruction() local
|
| H A D | indexed_array_analysis.cc | 482 const Shape& source_shape = operand->source()->shape(); in ReshapeToRemoveDegenerateDims() local
|
| H A D | dynamic_dimension_inference_test.cc | 997 auto source_shape = ShapeUtil::MakeShape(F32, {2, 2, 2}); in TEST_F() local
|
| H A D | hlo_sharding_util.cc | 355 std::optional<HloSharding> ReshapeSharding(const Shape& source_shape, in ReshapeSharding()
|
| H A D | shape_inference.cc | 2353 const Window& window, const Shape& source_shape, in InferSelectAndScatterShape()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | literal.cc | 202 const Shape& source_shape, const Shape& dest_shape, in StrideConfig()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 1490 auto source_shape = op.source().getType().cast<ShapedType>(); in InferShapeForXlaSelectAndScatterOp() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 7223 … def helper(shape, dim, index, source_shape, alpha, x_dtype=torch.float32, idx_dtype=torch.int32): argument
|