Searched defs:bc_shape (Results 1 – 3 of 3) sorted by relevance
13 std::vector<int64_t> bc_shape(elements.dim(), 1); in isin_default_kernel_gpu() local
330 TensorShape bc_shape = TensorShape::broadcast_shape(infos.tensor_shape()...); in broadcast_shape_and_valid_region() local
213 TensorShape bc_shape; in broadcast_shape() local