Searched defs:inputDim (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/cpp/c10d/ |
| H A D | ProcessGroupNCCLTest.cpp | 78 int inputDim = 3) in NCCLTest()
247 int inputDim) in SparseAllreduceNCCLTest()
429 const int inputDim = 3; in testSparseAllreduce() local
480 const int inputDim = 2500; in testSparseAllreduceLarge() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
| H A D | lhlo_elemental_utils.cc | 137 auto inputDim = b->create<arith::AddIOp>( in elementalLower() local
168 int inputDim = std::distance(broadcastDimensions.begin(), it); in elementalLowerImplForBroadcastInDimOps() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MirrorPad.cpp | 200 const auto inputDim = getSizeOfDimension(inputShape, i); in eval() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/thlo/IR/ |
| H A D | thlo_ops.cc | 601 int64_t inputDim = inputDims[permutationRef[i]]; in verify() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3601 uint32_t inputDim[] = {2, 2, 2, 2}; in normalizationOpTest() local
3633 uint32_t inputDim[] = {2, 2, 2, 6}; in localResponseNormOpTest() local
3689 uint32_t inputDim[] = {3, 3, 3}; in sliceTest() local
4306 uint32_t inputDim[] = {1, 2, 2, 1}, outputDim[] = {1, 1, 1, 1}; in resizeNearestNeighborTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| H A D | StablehloOps.cpp | 4724 static int64_t inferSliceDim(int64_t inputDim, int64_t start, int64_t end, in inferSliceDim()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| H A D | hlo_ops.cc | 6796 static int64_t inferSliceDim(int64_t inputDim, int64_t start, int64_t end, in inferSliceDim()
|