Searched defs:tensor_strides (Results 1 – 8 of 8) sorted by relevance
64 IntArrayRef tensor_strides, in inferExpandGeometryImpl()118 IntArrayRef tensor_strides, in inferExpandGeometry()127 IntArrayRef tensor_strides, in inferExpandGeometry_dimvector()148 std::vector<int64_t> infer_dense_strides(IntArrayRef tensor_sizes, IntArrayRef tensor_strides) { in infer_dense_strides()
63 IntArrayRef tensor_strides = tensor.strides(); in prepare_matrix_for_cublas() local82 IntArrayRef tensor_strides = tensor.strides(); in prepare_matrix_for_cublas() local128 IntArrayRef tensor_strides = tensor.strides(); in prepare_batch_matrix_for_cublas() local
230 auto tensor_strides = internal::strides<Layout>(tensor_dims); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local278 auto tensor_strides = internal::strides<Layout>(tensor_dims); in test_block_io_copy_using_reordered_dimensions_squeeze() local
104 const DSizes<Index, NumDims>& tensor_strides, in UpdateCoeffSet()
195 const auto tensor_strides = tensor.strides(); in align_to() local285 auto tensor_strides = tensor.strides(); in align_to() local
194 std::vector<IntArrayRef> tensor_strides = in get_stride_for_index() local
89 const Tensor& tensor_strides = tensor->get_nested_strides(); in is_safe_to_get_storage_as_tensor() local
1144 def significant_strides_equal(shape, meta_strides, tensor_strides): argument