Searched defs:dst_indices (Results 1 – 4 of 4) sorted by relevance
66 xla::XlaOp dst_indices = in Compile() local161 std::vector<int32> dst_indices(dim0); in Compile() local
138 std::vector<int32> dst_indices(attrs_.x_dim_count); in Convert() local
251 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices; in launch() local448 Eigen::DSizes<Eigen::DenseIndex, 5> dst_indices; in launch() local
1701 std::vector<int> dst_indices; in ReplaceResourceArgsWithVarHandleOps() local1753 std::vector<int> dst_indices; in ReplaceResourceArgsWithVarHandleOps() local1952 std::vector<int> dst_indices; in ReplaceAndPartitionXLAShardingVariable() local