Searched refs:bdims (Results 1 – 3 of 3) sorted by relevance
69 absl::Span<const int64> bdims = dim_sizes; in Compile() local70 bdims.remove_prefix(1); in Compile()71 cond_handle = xla::Broadcast(cond_handle, bdims); in Compile()
456 std::pair<int, int> bdims = spec.left_side ? std::make_pair(spec.m, spec.n) in XLA_TEST_P() local458 Array2D<float> bvals(bdims.first, bdims.second); in XLA_TEST_P()
613 std::vector<int64> bdims(dims.begin() + skip, dims.end()); in BroadcastableDims() local618 for (int64& dim : bdims) { in BroadcastableDims()626 dims.swap(bdims); in BroadcastableDims()628 return {dims, bdims}; in BroadcastableDims()