Searched defs:all_dims (Results 1 – 11 of 11) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | SortingCommon.cuh | 144 int all_dims = self_info.dims; in run_launcher() local
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesReduceOps.cpp | 137 auto all_dims = range(0, std::max((int64_t)1, logical_dim)); in boxed_reduction_batch_rule() local
156 auto all_dims = range(0, self.dim() - 1); in boxed_reduction_batch_rule() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | ReduceOps.cpp | 182 auto all_dims = std::vector<int64_t>(self.dim()); in is_std_inner_dim_fast_path() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 463 std::vector<int64_t> all_dims(operand.base_shape().rank()); in PartitionGatherTrivialIndexedOperandDimension() local
933 std::vector<int64_t> all_dims(indices.base_shape().rank()); in PartitionScatterIndexPassthroughPartition() local
|
| D | spmd_partitioner.cc | 1171 std::vector<int64_t> all_dims(shape.rank()); in Replicate() local
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | argument_spec.h | 244 int32_t all_dims = 0; in CompleteArgumentSpec() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 184 auto all_dims = ops::Const(root.WithOpName("all_dims"), {0}, {1}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 426 std::vector<int64_t> all_dims(operand->shape().rank()); in MakeReduceHlo() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 157 auto all_dims = Range(scope, Const(scope, 0), Rank(scope, grad_inputs[0]), in FillGrad() local
|
| /external/pytorch/functorch/csrc/dim/ |
| D | dim.cpp | 1229 Slice<mpy::hdl<Dim>> all_dims; in run_torch_function() local
2765 bool all_dims = true; in py_split() local
|
| /external/pytorch/aten/src/ATen/test/ |
| D | vulkan_api_test.cpp | 6721 std::vector<std::vector<int64_t>> all_dims; in TEST_F() local
6745 std::vector<std::vector<int64_t>> all_dims; in TEST_F() local
6786 std::vector<std::vector<int64_t>> all_dims; in TEST_F() local
|