Searched defs:dim_map (Results 1 – 5 of 5) sorted by relevance
52 MakeOperandStartIndexPermutations(absl::Span<const int64_t> dim_map, in MakeOperandStartIndexPermutations()
1080 std::vector<int64_t>& dim_map, bool check_lhs) { in CanPropagate()1982 std::vector<int64_t> dim_map(NumMappedDims()); in Propagate() local2725 std::vector<int64_t> dim_map(NumMappedDims()); in PropagateOnConv() local2973 auto dim_map = instr_to_dim_map_[producer]; in PropagateOnConstant() local3441 std::vector<int64_t> dim_map(NumMappedDims()); in PropagateOnBackpropFilterConv() local3824 std::vector<int64_t> dim_map(NumMappedDims()); in PerformSpaceToBatchOnConvolution() local
1266 auto dim_map = ComputeBitcastDimMap(bitcast->shape(), copy->shape()); in SwapCopyBitcastCopy() local
187 DSizes<int, NumDims> dim_map; in test_block_io_copy_using_reordered_dimensions() local205 DSizes<int, NumDims> dim_map; in test_block_io_copy_using_reordered_dimensions() local
1519 absl::InlinedVector<int64_t, 8> dim_map; in AlignLayouts() local