Searched defs:tensor_dim (Results 1 – 10 of 10) sorted by relevance
93 int64_t tensor_dim = sizes.size(1); in chunk_nested_tensor() local148 int64_t tensor_dim = sizes.size(1); in split_with_sizes_nested() local
196 const auto tensor_dim = tensor.sizes().size(); in align_to() local241 auto setNewSizesStridesNamesFor = [&](int64_t out_dim, int64_t tensor_dim) { in align_to()
118 auto tensor_dim = tensor.dim(); in makeBatched() local
67 int64_t tensor_dim = static_cast<int64_t>(tensor_sizes.size()); in inferExpandGeometryImpl() local
169 int64_t tensor_dim = 0; in alignBatchDimsAtFront() local
55 void check_names_valid_for(size_t tensor_dim, DimnameList names) { in check_names_valid_for()
217 auto tensor_dim = sizes.size(); in is_onednn_matmul_strides() local
685 const int64_t tensor_dim = sizes[dim_index]; in __anonc321d2750602() local
768 int64_t tensor_dim = TFE_TensorHandleDim(component.get(), i, status); in VerifyPackTensorShapeAndDtype() local
1901 const auto tensor_dim = tensor->getDimensions(); in ConvertConv2DHelper() local2859 const auto tensor_dim = tensor->getDimensions(); in ConvertConv3DHelper() local