Searched refs:res_shape (Results 1 – 3 of 3) sorted by relevance
214 res_shape = _get_flattern_shape(output_shape)215 res = tik_instance.Tensor(dtype, res_shape, name="res", scope=tik.scope_gm)
558 auto res_shape = ShapeJoin(shape(), other_undetermined_tensor->shape()); in Join() local559 if (res_shape == nullptr) { in Join()565 return std::make_shared<AbstractUndetermined>(element, res_shape); in Join()577 auto res_shape = ShapeJoin(this->shape(), other_tensor->shape()); in Join() local578 if (res_shape == nullptr) { in Join()584 return std::make_shared<AbstractTensor>(element, res_shape); in Join()
385 res_shape = broadcast_shape387 res_shape += data_shape[tuple_index_len:]388 res = const_utils.make_tensor([], data.dtype, res_shape)