Searched refs:merged_shape (Results 1 – 3 of 3) sorted by relevance
69 TensorShape merged_shape(a_shape); in Compute() local70 if (b_shape.dims() > 0) merged_shape = b_shape; in Compute()71 if (x_shape.dims() > 0) merged_shape = x_shape; in Compute()74 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, merged_shape, &output)); in Compute()83 auto merged_shape_vec = BCast::FromShape(merged_shape); in Compute()
676 merged_shape = shape_list[0]678 merged_shape.merge_with(s)679 return merged_shape.as_list()
1459 ShapeHandle merged_shape; in __anondb9326b22302() local1461 c->Merge(desired_shape_handle, input_shape_handle, &merged_shape)); in __anondb9326b22302()1462 c->set_output(0, merged_shape); in __anondb9326b22302()