Searched defs:shape_i (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | array_grad.cc | 149 std::vector<string> shape_i; in ConcatGradHelper() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 1682 def _test_spadd_shape(nnz, shape_i, shape_v=None): argument
1858 def _test_basic_ops_shape(self, nnz_x1, nnz_x2, shape_i, shape_v, dtype, device, coalesced): argument
1991 def _test_sparse_mask_shape(self, nnz_x1, nnz_x2, shape_i, shape_v, dtype, device, coalesced): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | einsum.cc | 490 const int64_t shape_i = shape[std::get<I>(index_tuple)]; in ProdShapeWithIndexInTuple() local
|