Searched defs:return_indices (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | merge_control_flow.cc | 438 llvm::SmallVector<llvm::SmallVector<int, 4>>& return_indices) { in ReplaceExternalUsage()
481 llvm::SmallVector<llvm::SmallVector<int, 4>>& return_indices, in CreateYieldOps()
529 auto return_indices = GetReturnIndicesVec(if_op_segment); in MergeIfPerSegment() local
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/ |
| H A D | functional.py | 484 return_indices=False, argument
515 return_indices=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | meta_spmd_expander.cc | 1173 std::vector<int> return_indices; in ExpandOp() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_mps.py | 2237 …def helper(shape, ks, padding=0, dilation=1, ceil_mode=False, return_indices=False, test_ties=Fals… argument
6882 def helper(input_shape, out_shape, return_indices, dtype, channels_last=False): argument
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_opset9.py | 1433 def _max_pool(name, tuple_fn, ndims, return_indices): argument
|