Searched defs:bcast_shape (Results 1 – 4 of 4) sorted by relevance
362 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local479 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
769 TensorShape bcast_shape = BCast::ToShape(bcast.output_shape()); in Compute() local
1371 SmallVector<int64_t, 4> bcast_shape(input_rank, 1); in convertEluOp() local
1063 Shape bcast_shape = ShapeUtil::ChangeElementType(shape, PRED); in Broadcast() local