Searched defs:dim_map (Results 1 – 9 of 9) sorted by relevance
329 int32_t* dim_map, in compute_dim_map()400 int32_t* dim_map, in get_in_coord()450 int32_t* dim_map, in get_in_ix()
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
52 MakeOperandStartIndexPermutations(absl::Span<const int64_t> dim_map, in MakeOperandStartIndexPermutations()
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
80 int32_t dim_map[kTensorDimensionLimit]; in index_Tensor_out() local
106 int32_t dim_map[kTensorDimensionLimit]; in index_put_out() local
136 def dim_map(self) -> List[int]: member in DTensorSpec
1519 absl::InlinedVector<int64_t, 8> dim_map; in AlignLayouts() local